From aed95319411b8b5190b5b418b65b523b1436bcfd Mon Sep 17 00:00:00 2001
From: Robin Manhaeve <robin.manhaeve@cs.kuleuven.be>
Date: Fri, 26 Jan 2024 13:21:34 +0100
Subject: [PATCH] Updated requirements.txt and setup.py

---
 requirements.txt | 12 ++++++------
 setup.py         | 16 +++++++++-------
 2 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index a86b2db..e86bd74 100644
--- 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
\ No newline at end of file
+torch
+torchvision
+numpy
+pandas
+pyparsing
+dgl
diff --git a/setup.py b/setup.py
index beaeb9b..00630bb 100644
--- 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",
     ],
-)
\ No newline at end of file
+)