Merge pull request #2 from konstin/main

WIP: Add dependencies for inference to setup.py
amyxlu · Oct 27, 2020 · 565be9d · 565be9d
2 parents df1ad11 + 43f5ed6
commit 565be9d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -15,5 +15,13 @@
         "License :: OSI Approved :: MIT License",
         "Operating System :: Linux",
     ],
-    python_requires='>=3.6',
+    python_requires=">=3.6",
+    install_requires=[
+        "numpy>=1.19,<2.0",
+        "torch>=1.0,<2.0",
+        "lmdb>=1.0,<2.0",
+        # "tape-proteins>=0.4,<=1.0",
+        "tape-proteins @ git+https://github.com/konstin/tape@patch-1",
+        "scipy>=1.5,<2.0",
+    ],
 )