add install_requires

CSDUlm · May 31, 2022 · cdbfe7f · cdbfe7f
1 parent 1966dee
commit cdbfe7f
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -7,12 +7,26 @@
 
 setup(
     name="wsingular",
-    version="0.1.1",
+    version="0.1.2",
     description="Wasserstein Singular Vectors",
     long_description=long_description,
     long_description_content_type="text/markdown",
     author="Geert-Jan Huizing, Laura Cantini, Gabriel Peyré",
     url="https://github.com/gjhuizing/wsingular",
     author_email="huizing@ens.fr",
     packages=["wsingular"],
+    install_requires=[
+        "torch",
+        "protobuf<=3.20.1",
+        "tensorboard",
+        "pandas",
+        "sklearn",
+        "seaborn",
+        "matplotlib",
+        "scipy",
+        "tqdm",
+        "numpy",
+        "pot>=0.8",
+        "scikit-network",
+    ],
 )