Skip to content

Commit 66afa0c

Browse files
author
Yadan Wei
committed
fix typo
1 parent bd6e4bb commit 66afa0c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/dlc_tests/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -905,7 +905,7 @@ def skip_torchdata_test(request):
905905
if not image_uri:
906906
return
907907

908-
skip_dict = {">2.1.1": ["cpu", "cu118", "cu121"], ">=2.4": ["cpu", "cu124"]}
908+
skip_dict = {">2.1.1": ["cpu", "cu118", "cu121"], ">=2.4,<2.6": ["cpu", "cu124"]}
909909
if _validate_pytorch_framework_version(request, image_uri, "skip_torchdata_test", skip_dict):
910910
pytest.skip(
911911
f"Torchdata has paused development as of July 2023 and the latest compatible PyTorch version is 2.1.1."
@@ -925,7 +925,7 @@ def skip_smdebug_v1_test(request):
925925
else:
926926
return
927927

928-
skip_dict = {"==2.0.*": ["cu121"], ">=2.1,<2.4": ["cpu", "cu121"], ">=2.4,<2.6": ["cpu", "cu124"]}
928+
skip_dict = {"==2.0.*": ["cu121"], ">=2.1,<2.4": ["cpu", "cu121"], ">=2.4": ["cpu", "cu124"]}
929929
if _validate_pytorch_framework_version(request, image_uri, "skip_smdebug_v1_test", skip_dict):
930930
pytest.skip(f"SM Profiler v1 is on path for deprecation, skipping test")
931931

0 commit comments

Comments
 (0)