From 43f5ed6a66aa2400298c19f8c3efdbe22f3cdbb0 Mon Sep 17 00:00:00 2001
From: konstin <konstin@mailbox.org>
Date: Thu, 22 Oct 2020 11:37:50 +0200
Subject: [PATCH] WIP: Add dependencies for inference to setup.py

---
 setup.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 2e0cdeb..a4e8c68 100644
--- 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",
+    ],
 )