We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4c5cd25 commit 5db1bd7Copy full SHA for 5db1bd7
nequip/__init__.py
@@ -11,10 +11,8 @@
11
torch_version = packaging.version.parse(torch.__version__)
12
13
# 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")
+assert (torch_version == packaging.version.parse("1.11")) or (
+ torch_version >= packaging.version.parse("1.13")
18
), f"NequIP supports PyTorch 1.11.* or 1.13.* or later, but {torch_version} found"
19
20
# warn if using 1.13* or 2.0.*
0 commit comments