Merge pull request #467 from mir-group/versioncheck

support +cpu etc versions, fixes #459
mir-group · Oct 21, 2024 · 1d672a7 · 1d672a7
2 parents d3a7763 + f9d55e0
commit 1d672a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 (