Specify requirements for installation

JetBrains-Research · Aug 18, 2021 · 88b1a32 · 88b1a32
1 parent 71d4023
commit 88b1a32
Showing 1 changed file with 9 additions and 3 deletions.
diff --git 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",