Skip to content

Commit 30370fb

Browse files
daniellutzjiridanek
authored andcommitted
RHAIENG-1512: fix(make/test): PyTorch ROCm 6.3 tests (#2597)
1 parent 9dc38fb commit 30370fb

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)