changing to correct smp version in unit test

aws · Jul 24, 2024 · 6b0c115 · 6b0c115
1 parent d83851a
commit 6b0c115
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/sagemaker/image_uris/test_smp_v2.py b/tests/unit/sagemaker/image_uris/test_smp_v2.py
@@ -38,7 +38,7 @@ def test_smp_v2(load_config):
                         if "2.1" in version or "2.2" in version or "2.3" in version:
                             cuda_vers = "cu121"
 
-                        if "2.3" in version:
+                        if "2.4" in version:
                             py_version = "py311"
 
                         uri = image_uris.get_training_image_uri(