release torch<2.6 restriction

florian-huber · florian-huber · commit 8da42247a434 · 2025-11-07T14:47:32.000+01:00
diff --git a/setup.py b/setup.py
@@ -34,13 +34,13 @@
         "pandas",
         "scikit-learn",
         "tensorboard",
-        "torch<2.6",
+        "torch>=2.4",
         "tqdm",
         "matplotlib>=3.8.3"
     ],
     extras_require={"train": ["rdkit"],
                     "dev": ["bump2version",
-                            "isort>=4.2.5,<5",
+                            "isort>=4.2.5",
                             "ruff",
                             "pytest",
                             "pytest-cov",
