chore: add ranking name argument to get_top_ranked_config_name

Author: evakravi

src/sagemaker/jumpstart/exceptions.py

@@ -150,7 +150,6 @@ def __init__(
                 model. (Default: None).
 
         """
-        version = version or "*"
         if message:
             self.message = message
         else:
@@ -199,7 +198,6 @@ def __init__(
         version: Optional[str] = None,
         message: Optional[str] = None,
     ):
-        version = version or "*"
         if message:
             self.message = message
         else:

src/sagemaker/jumpstart/utils.py

@@ -1148,6 +1148,7 @@ def get_top_ranked_config_name(
     tolerate_deprecated_model: bool = False,
     tolerate_vulnerable_model: bool = False,
     hub_arn: Optional[str] = None,
+    ranking_name: enums.JumpStartConfigRankingName = enums.JumpStartConfigRankingName.DEFAULT,
 ) -> Optional[str]:
     """Returns the top ranked config name for the given model ID and region.
 
@@ -1168,13 +1169,17 @@ def get_top_ranked_config_name(
 
     if scope == enums.JumpStartScriptScope.INFERENCE:
         return (
-            model_specs.inference_configs.get_top_config_from_ranking().config_name
+            model_specs.inference_configs.get_top_config_from_ranking(
+                ranking_name=ranking_name
+            ).config_name
             if model_specs.inference_configs
             else None
         )
     if scope == enums.JumpStartScriptScope.TRAINING:
         return (
-            model_specs.training_configs.get_top_config_from_ranking().config_name
+            model_specs.training_configs.get_top_config_from_ranking(
+                ranking_name=ranking_name
+            ).config_name
             if model_specs.training_configs
             else None
         )