Skip to content

Commit 6b0c115

Browse files
committed
changing to correct smp version in unit test
1 parent d83851a commit 6b0c115

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unit/sagemaker/image_uris/test_smp_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def test_smp_v2(load_config):
3838
if "2.1" in version or "2.2" in version or "2.3" in version:
3939
cuda_vers = "cu121"
4040

41-
if "2.3" in version:
41+
if "2.4" in version:
4242
py_version = "py311"
4343

4444
uri = image_uris.get_training_image_uri(

0 commit comments

Comments
 (0)