From 88b1a3208c675c21abe01b59dc430a120d2d7a06 Mon Sep 17 00:00:00 2001 From: Egor Spirin Date: Wed, 18 Aug 2021 13:04:38 +0500 Subject: [PATCH] Specify requirements for installation --- setup.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index e11483b..30e89bf 100644 --- a/setup.py +++ b/setup.py @@ -1,12 +1,18 @@ from setuptools import setup, find_packages -VERSION = "0.0.3" +VERSION = "0.0.4" with open("README.md") as readme_file: readme = readme_file.read() -with open("requirements.txt") as requirements_file: - install_requires = [line.strip() for line in requirements_file] +install_requires = [ + "torch~=1.7.1", + "tqdm~=4.58.0", + "numpy~=1.20.1", + "pytorch-lightning==1.1.7", + "wandb~=0.10.20", + "hydra-core~=1.0.6", +] setup_args = dict( name="code2seq",