Skip to content

Commit 323d567

Browse files
committed
Fix PyTorch ROCm 6.3 tests
1 parent 1237866 commit 323d567

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -277,8 +277,8 @@ def test_image_pyprojects_version_alignment(subtests: pytest_subtests.plugin.Sub
277277
("setuptools", ("~=80.9.0", "==80.9.0")),
278278
("wheel", ("==0.45.1", "~=0.45.1")),
279279
("tensorboard", ("~=2.17.0", "~=2.20.0")),
280-
("torch", ("==2.7.1", "==2.7.1+cu128", "==2.7.1+rocm6.2.4")),
281-
("torchvision", ("==0.22.1", "~=0.22.1", "==0.22.1+cu128", "==0.22.1+rocm6.2.4")),
280+
("torch", ("==2.7.1", "==2.7.1+cu128", "==2.7.1+rocm6.3")),
281+
("torchvision", ("==0.22.1", "~=0.22.1", "==0.22.1+cu128", "==0.22.1+rocm6.3")),
282282
(
283283
"matplotlib",
284284
("~=3.10.6",),

0 commit comments

Comments
 (0)