fix: bug in get latest version was getting the max sorted alphabetically

instead of sem-ver

Author: Eli Davidson

src/sagemaker/jumpstart/cache.py

@@ -540,9 +540,7 @@ def _select_version(
         """
 
         if version_str == "*":
-            if len(available_versions) == 0:
-                return None
-            return str(max(available_versions))
+            return utils.get_latest_version(available_versions)
 
         if model_type == JumpStartModelType.PROPRIETARY:
             if "*" in version_str:

src/sagemaker/jumpstart/utils.py

@@ -1630,3 +1630,7 @@ def get_draft_model_content_bucket(provider: Dict, region: str) -> str:
             return get_jumpstart_gated_content_bucket(region=region)
         return get_jumpstart_content_bucket(region=region)
     return neo_bucket
+
+def get_latest_version(versions: List[str]) -> Optional[str]:
+    return None if not versions else max(versions, key=Version)
+

tests/unit/sagemaker/jumpstart/test_utils.py

@@ -2143,6 +2143,14 @@ def test_add_instance_rate_stats_to_benchmark_metrics_client_ex(
 def test_has_instance_rate_stat(stats, expected):
     assert utils.has_instance_rate_stat(stats) is expected
 
+def test_get_latest_version():
+    assert utils.get_latest_version(["2.9.1", "2.16.0", "1.0.0"]) == "2.16.0"
+
+def test_get_latest_version_empty_list_is_none():
+    assert utils.get_latest_version([]) == None
+
+def test_get_latest_version_none_is_none():
+    assert utils.get_latest_version(None) == None
 
 @pytest.mark.parametrize(
     "data, expected",