torch and dicee

dice-group · Jun 27, 2024 · d85534b · d85534b
1 parent bdf2b6f
commit d85534b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
     "matplotlib>=3.3.4",
     "scikit-learn>=1.4.1",
     "owlready2>=0.40",
-    "torch>=1.7.1,<2.2.0",
+    "torch==2.2.0",
     "rdflib>=6.0.2",
     "pandas>=1.5.0",
     "sortedcontainers>=2.4.0",  # AB: This is required
@@ -25,6 +25,7 @@
     "transformers>=4.38.1",
     "pytest>=7.2.2",
     "owlapy==1.1.0",
+    "dicee==0.1.4",
     "dicee>=0.1.2",
     "ontosample>=0.2.2",
     "sphinx>=7.2.6",