fix: Minor fixes

Author: chrstfu

src/sagemaker/jumpstart/utils.py

@@ -863,7 +863,9 @@ def validate_model_id_and_get_type(
             model_version=model_version,
             sagemaker_session=sagemaker_session,
         )
-        return model_types[0]  # Currently this function only supports one model type
+        return (
+            model_types[0] if model_types else None
+        )  # Currently this function only supports one model type
 
     s3_client = sagemaker_session.s3_client if sagemaker_session else None
     region = region or constants.JUMPSTART_DEFAULT_REGION_NAME
@@ -908,8 +910,8 @@ def _validate_hub_service_model_id_and_get_type(
     )
 
     hub_content_model_types = []
-    model_types_field = getattr(hub_model_specs, "model_types", [])
-    model_types = model_types_field if model_types_field is not None else []
+    model_types_field: Optional[List[str]] = getattr(hub_model_specs, "model_types", [])
+    model_types = model_types_field if model_types_field else []
     for model_type in model_types:
         try:
             hub_content_model_types.append(enums.JumpStartModelType[model_type])

tests/unit/sagemaker/jumpstart/hub/test_marketplace_hub_content.py

@@ -98,12 +98,14 @@ def _test_proprietary_model(input_version, expected_version, sagemaker_session):
 
 
 @pytest.mark.parametrize(
-    "get_model_specs_response, expected, expected_exception, expected_message",
+    "get_model_specs_attr, get_model_specs_response, expected, expected_exception, expected_message",
     [
-        (None, [], None, None),
-        ([], [], None, None),
-        (["OPEN_WEIGHTS"], [JumpStartModelType.OPEN_WEIGHTS], None, None),
+        (False, None, [], None, None),
+        (True, None, [], None, None),
+        (True, [], [], None, None),
+        (True, ["OPEN_WEIGHTS"], [JumpStartModelType.OPEN_WEIGHTS], None, None),
         (
+            True,
             ["OPEN_WEIGHTS", "PROPRIETARY"],
             [JumpStartModelType.OPEN_WEIGHTS, JumpStartModelType.PROPRIETARY],
             None,
@@ -113,10 +115,15 @@ def _test_proprietary_model(input_version, expected_version, sagemaker_session):
 )
 @patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor.get_model_specs")
 def test_validate_hub_service_model_id_and_get_type(
-    mock_get_model_specs, get_model_specs_response, expected, expected_exception, expected_message
+    mock_get_model_specs,
+    get_model_specs_attr,
+    get_model_specs_response,
+    expected,
+    expected_exception,
+    expected_message,
 ):
     mock_object = MagicMock()
-    if get_model_specs_response:
+    if get_model_specs_attr:
         mock_object.model_types = get_model_specs_response
     mock_get_model_specs.return_value = mock_object