setup

killiansheriff · Mar 14, 2024 · 2872286 · 2872286
1 parent bfa6d3e
commit 2872286
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 import os
 
-from setuptools import find_packages, setup
+from setuptools import setup
 
 # Get description from README
 root = os.path.abspath(os.path.dirname(__file__))
@@ -26,14 +26,13 @@
         "pandas",
         "scikit_learn",
         "scipy",
-        "simplex",
-        #"torch==2.0.1",
         "torch_geometric",
-        #"torch_scatter==2.1.2",
         "tqdm",
         "NshellFinder",
         "polyaenum",
         'nsimplex',
+        "torch==2.0.1",
+        "torch_scatter==2.1.2",
     ],
     classifiers=[],
 )