Skip to content

Commit 8fce327

Browse files
committed
fix confest for sm tests
1 parent 0a1ce87 commit 8fce327

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/sagemaker_tests/pytorch/training/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -475,7 +475,7 @@ def skip_smdmodelparallel_test(
475475
ecr_image,
476476
):
477477
skip_dict = {
478-
"==1.13.*": ["cpu", "cu117"],
478+
"==2.0.*": ["cu121"],
479479
">=2.1,<2.4": ["cpu", "cu121"],
480480
">=2.4,<2.6": ["cpu", "cu124"],
481481
">=2.6,<2.7.1": ["cpu", "cu126"],

0 commit comments

Comments
 (0)