From 0b3571dfefee782c46c9526446bdc44bc5d3630e Mon Sep 17 00:00:00 2001 From: ben rhodes Date: Thu, 19 Dec 2024 13:13:05 +0000 Subject: [PATCH] bound torch version from above --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 8412df0..ea91aa3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -20,7 +20,7 @@ dependencies = [ "ase>=3.23.0", "numpy>=1.26.4", "scipy>=1.13.1", - "torch>=2.2.0", + "torch>=2.2.0, <3.0.0", "dm-tree>=0.1.8", "tqdm>=4.66.5", ]