Updated requirements.txt and setup.py

ML-KULeuven · Jan 26, 2024 · aed9531 · aed9531
1 parent bde6d05
commit aed9531
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 13 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
-torch~=1.5.1
-torchvision~=0.6.1
-numpy~=1.18.1
-pandas~=1.2.4
-pyparsing~=2.4.7
-dgl~=0.6.1
+torch
+torchvision
+numpy
+pandas
+pyparsing
+dgl
diff --git a/setup.py b/setup.py
@@ -14,12 +14,14 @@
     long_description_content_type="text/markdown",
     packages=setuptools.find_packages(include=["deepstochlog", "deepstochlog.*"]),
     python_requires='>=3',
+    package_data={'deepstochlog': ['*.pl']},
+    include_package_data=True,
     install_requires=[
-        "torch~=1.5.1",
-        "torchvision~=0.6.1",
-        "numpy~=1.18.1",
-        "pandas~=1.2.4",
-        "pyparsing~=2.4.7",
-        "dgl~=0.6.1",
+        "torch",
+        "torchvision",
+        "numpy",
+        "pandas",
+        "pyparsing",
+        "dgl",
     ],
-)
+)