fix codestyles

Author: benieric

src/sagemaker/jumpstart/hub/utils.py

@@ -195,31 +195,6 @@ def is_gated_bucket(bucket_name: str) -> bool:
     return bucket_name in constants.JUMPSTART_GATED_BUCKET_NAME_SET
 
 
-def get_hub_model_version(
-    hub_name: str,
-    hub_model_name: str,
-    hub_model_type: str,
-    hub_model_version: Optional[str] = None,
-    sagemaker_session: Session = constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
-) -> str:
-    """Returns available Jumpstart hub model version.
-
-    It will attempt both a semantic HubContent version search and Marketplace version search.
-    If the Marketplace version is also semantic, this function will default to HubContent version.
-
-    Raises:
-        ClientError: If the specified model is not found in the hub.
-        KeyError: If the specified model version is not found.
-    """
-
-    try:
-        hub_content_summaries = sagemaker_session.list_hub_content_versions(
-            hub_name=hub_name, hub_content_name=hub_model_name, hub_content_type=hub_model_type
-        ).get("HubContentSummaries")
-    except Exception as ex:
-        raise Exception(f"Failed calling list_hub_content_versions: {str(ex)}")
-
-
 def get_hub_model_version(
     hub_name: str,
     hub_model_name: str,

tests/integ/sagemaker/jumpstart/utils.py

@@ -66,6 +66,7 @@ def get_sm_session_with_override() -> Session:
         sagemaker_client=sagemaker,
     )
 
+
 def get_training_dataset_for_model_and_version(model_id: str, version: str) -> dict:
     return TRAINING_DATASET_MODEL_DICT[(model_id, version)]