From 767c68af479618c46ab973733042b89eb5a09414 Mon Sep 17 00:00:00 2001
From: Stefan Doerr <stefdoerr@gmail.com>
Date: Thu, 20 Feb 2025 12:55:16 +0200
Subject: [PATCH] support pytorch 2.6

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 3f217596..dd7d3467 100644
--- 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"