Skip to content

Commit 5db1bd7

Browse files
fix version check
1 parent 4c5cd25 commit 5db1bd7

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

nequip/__init__.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,8 @@
1111
torch_version = packaging.version.parse(torch.__version__)
1212

1313
# only allow 1.11*, 1.13* or higher (no 1.12.*)
14-
assert (torch_version > packaging.version.parse("1.11.0")) and not (
15-
packaging.version.parse("1.12.0")
16-
<= torch_version
17-
< packaging.version.parse("1.13.0")
14+
assert (torch_version == packaging.version.parse("1.11")) or (
15+
torch_version >= packaging.version.parse("1.13")
1816
), f"NequIP supports PyTorch 1.11.* or 1.13.* or later, but {torch_version} found"
1917

2018
# warn if using 1.13* or 2.0.*

0 commit comments

Comments
 (0)