address nits

Author: Malav Shastri

src/sagemaker/jumpstart/accessors.py

@@ -301,7 +301,7 @@ def get_model_specs(
 
             except Exception as ex:
                 logging.info(
-                    "Recieved expection while calling APIs for ContentType Model: " + str(ex)
+                    "Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: " + str(ex)
                 )
                 hub_model_arn = construct_hub_model_arn_from_inputs(
                     hub_arn=hub_arn, model_name=model_id, version=version

src/sagemaker/jumpstart/hub/hub.py

@@ -286,7 +286,7 @@ def describe_model(
             )
 
         except Exception as ex:
-            logging.info("Recieved expection while calling APIs for ContentType Model: " + str(ex))
+            logging.info("Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: " + str(ex))
             model_version = get_hub_model_version(
                 hub_model_name=model_name,
                 hub_model_type=HubContentType.MODEL_REFERENCE.value,

src/sagemaker/jumpstart/hub/utils.py

@@ -193,7 +193,11 @@ def get_hub_model_version(
     hub_model_version: Optional[str] = None,
     sagemaker_session: Session = constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
 ) -> str:
-    """Returns available Jumpstart hub model version"""
+    """Returns available Jumpstart hub model version
+
+    Raises:
+        ResourceNotFound: If the specified model is not found in the hub.
+    """
 
     try:
         hub_content_summaries = sagemaker_session.list_hub_content_versions(

src/sagemaker/jumpstart/types.py

@@ -17,6 +17,7 @@
 from copy import deepcopy
 from enum import Enum
 from typing import Any, Dict, List, Optional, Set, Union
+from sagemaker.jumpstart.constants import HUB_ARN_REGEX, HUB_CONTENT_ARN_REGEX
 from sagemaker.model_card.model_card import ModelCard, ModelPackageModelCard
 from sagemaker.utils import get_instance_type_family, format_tags, Tags, deep_override_dict
 from sagemaker.model_metrics import ModelMetrics
@@ -1412,6 +1413,7 @@ def __init__(self, spec: Dict[str, Any], is_hub_content: Optional[bool] = False)
 
         Args:
             spec (Dict[str, Any]): Dictionary representation of spec.
+            is_hub_content (Optional[bool]): Whether the model is from a private hub.
         """
         super().__init__(spec, is_hub_content)
         self.from_json(spec)
@@ -1666,10 +1668,6 @@ def __init__(
     def extract_region_from_arn(arn: str) -> Optional[str]:
         """Extracts hub_name, content_name, and content_version from a HubContentArn"""
 
-        HUB_CONTENT_ARN_REGEX = (
-            r"arn:(.*?):sagemaker:(.*?):(.*?):hub-content/(.*?)/(.*?)/(.*?)/(.*?)$"
-        )
-        HUB_ARN_REGEX = r"arn:(.*?):sagemaker:(.*?):(.*?):hub/(.*?)$"
 
         match = re.match(HUB_CONTENT_ARN_REGEX, arn)
         hub_region = None