You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: .github/workflows/tests.yml
+5-5Lines changed: 5 additions & 5 deletions
Original file line number
Diff line number
Diff line change
@@ -53,7 +53,7 @@ jobs:
53
53
TORCH: "${{ matrix.torch-version }}"
54
54
run: |
55
55
# install packages that aren't required dependencies but that the tests do need
56
-
pip install h5py vesin
56
+
pip install h5py vesin packaging
57
57
#pip install schedulefree #Disabled to reduce CI runtime; uncomment to test ScheduleFreeLightningModule
58
58
# install CuEquivariance if CUDA is available
59
59
if python3 -c "import torch; exit(0 if torch.cuda.is_available() else 1)"; then
@@ -62,16 +62,16 @@ jobs:
62
62
else
63
63
echo "Skipping CuEquivariance installation (CUDA not available)"
64
64
fi
65
-
# install OpenEquivariance only if torch >= 2.4 and CUDA is available
66
-
if python3 -c "import torch; import packaging.version; exit(0 if packaging.version.parse(torch.__version__).base_version >= '2.4' else 1)"; then
65
+
# install OpenEquivariance only if torch >= 2.7 and CUDA is available
66
+
if python3 -c "import torch; import packaging.version; exit(0 if packaging.version.parse(torch.__version__) >= packaging.version.parse('2.7') else 1)"; then
67
67
if python3 -c "import torch; exit(0 if torch.cuda.is_available() else 1)"; then
68
-
echo "Installing OpenEquivariance (torch >= 2.4 and CUDA available)"
68
+
echo "Installing OpenEquivariance (torch >= 2.7 and CUDA available)"
69
69
pip install openequivariance
70
70
else
71
71
echo "Skipping OpenEquivariance installation (CUDA not available)"
0 commit comments