chore: improve logic for getting training instance variants

evakravi · evakravi · commit 97a581612f32 · 2023-10-26T12:11:15.000Z
diff --git a/src/sagemaker/jumpstart/artifacts/instance_types.py b/src/sagemaker/jumpstart/artifacts/instance_types.py
@@ -180,7 +180,7 @@ def _retrieve_instance_types(
                 training_instance_type
             )
             if training_instance_type is not None
-            and hasattr(model_specs, "training_instance_type_variants")
+            and getattr(model_specs, "training_instance_type_variants", None) is not None
             else []
         )
         instance_types = (

Original file line number	Diff line number	Diff line change
`@@ -180,7 +180,7 @@ def _retrieve_instance_types(`
`180`	`180`	`training_instance_type`
`181`	`181`	`)`
`182`	`182`	`if training_instance_type is not None`
`183`		`- and hasattr(model_specs, "training_instance_type_variants")`
	`183`	`+ and getattr(model_specs, "training_instance_type_variants", None) is not None`
`184`	`184`	`else []`
`185`	`185`	`)`
`186`	`186`	`instance_types = (`