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

@@ -1525,3 +1525,6 @@ def wrapped_f(*args, **kwargs):
     if _func is None:
         return wrapper_cache
     return wrapper_cache(_func)
+
+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

@@ -2142,6 +2142,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",