Skip to content

Commit 4fae2c7

Browse files
committed
update gh workflows for oeq test
1 parent aad7b1a commit 4fae2c7

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

.github/workflows/tests.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ jobs:
5353
TORCH: "${{ matrix.torch-version }}"
5454
run: |
5555
# install packages that aren't required dependencies but that the tests do need
56-
pip install h5py vesin
56+
pip install h5py vesin packaging
5757
#pip install schedulefree #Disabled to reduce CI runtime; uncomment to test ScheduleFreeLightningModule
5858
# install CuEquivariance if CUDA is available
5959
if python3 -c "import torch; exit(0 if torch.cuda.is_available() else 1)"; then
@@ -62,16 +62,16 @@ jobs:
6262
else
6363
echo "Skipping CuEquivariance installation (CUDA not available)"
6464
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
6767
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)"
6969
pip install openequivariance
7070
else
7171
echo "Skipping OpenEquivariance installation (CUDA not available)"
7272
fi
7373
else
74-
echo "Skipping OpenEquivariance installation (torch < 2.4)"
74+
echo "Skipping OpenEquivariance installation (torch < 2.7)"
7575
fi
7676
pip install --upgrade-strategy only-if-needed .
7777
pip install pytest pytest-xdist[psutil]

0 commit comments

Comments
 (0)