support pytorch 2.6

torchmd · Feb 20, 2025 · 767c68a · 767c68a
1 parent eea7673
commit 767c68a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ classifiers = [
 dependencies = [
   "h5py",
   # "nnpops",
-  "torch==2.5.1.*",
+  "torch>=2.5",
   "torch_geometric",
   "lightning",
   "tqdm",
@@ -34,5 +34,5 @@ include = ["torchmdnet*"]
 namespaces = false
 
 [build-system]
-requires = ["setuptools>=64", "setuptools-scm>=8", "torch==2.5.1.*"]
+requires = ["setuptools>=64", "setuptools-scm>=8", "torch>=2.5"]
 build-backend = "setuptools.build_meta"