specify minimum torch version (fixes #11 ) + remove tests from pip in…

…stall as this created mess (see #18)
cai4cai · Nov 30, 2022 · f25cb0b · f25cb0b
1 parent 4e5ae6d
commit f25cb0b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -22,9 +22,9 @@ def readme():
     author="CAI4CAI research group",
     author_email="[email protected]",
     license="Apache-2.0",
-    packages=setuptools.find_packages(),
+    packages=setuptools.find_packages(exclude=("tests",)),
     install_requires=[
-        "torch",
+        "torch>=1.13",
     ],
     extras_require={
         "extras": ["jax", "cupy"],