From f9d55e09ad0fe5c69dd9673b5d1d3caac134ec29 Mon Sep 17 00:00:00 2001 From: anjohan Date: Mon, 21 Oct 2024 13:50:01 -0600 Subject: [PATCH] support +cpu etc versions, should fix #459 --- nequip/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nequip/__init__.py b/nequip/__init__.py index ce145b41..2481bf56 100644 --- a/nequip/__init__.py +++ b/nequip/__init__.py @@ -8,7 +8,7 @@ import warnings # torch version checks -torch_version = packaging.version.parse(torch.__version__) +torch_version = packaging.version.parse(torch.__version__.split("+")[0]) # only allow 1.11*, 1.13* or higher (no 1.12.*) assert (torch_version == packaging.version.parse("1.11")) or (