Skip to content

Commit 3ca1deb

Browse files
committed
fix: Fixing integration tests
1 parent cd82335 commit 3ca1deb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/jumpstart/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -908,7 +908,9 @@ def _validate_hub_service_model_id_and_get_type(
908908
)
909909

910910
hub_content_model_types = []
911-
for model_type in getattr(hub_model_specs, "model_types", []):
911+
model_types_field = getattr(hub_model_specs, "model_types", [])
912+
model_types = model_types_field if model_types_field is not None else []
913+
for model_type in model_types:
912914
try:
913915
hub_content_model_types.append(enums.JumpStartModelType[model_type])
914916
except ValueError:

0 commit comments

Comments
 (0)