Relax set deployment error handling

Author: Jonathan Makunga

src/sagemaker/jumpstart/factory/model.py

@@ -674,11 +674,7 @@ def _add_config_name_to_init_kwargs(kwargs: JumpStartModelInitKwargs) -> JumpSta
         resolved_config = specs.inference_configs.configs[kwargs.config_name].resolved_config
         supported_instance_types = resolved_config.get("supported_inference_instance_types", [])
         if kwargs.instance_type not in supported_instance_types:
-            raise ValueError(
-                f"Instance type {kwargs.instance_type} "
-                f"is not supported for config {kwargs.config_name}."
-            )
-
+            JUMPSTART_LOGGER.warning("Overriding instance type to %s", kwargs.instance_type)
     return kwargs
 
 

tests/unit/sagemaker/jumpstart/model/test_model.py

@@ -1941,12 +1941,6 @@ def test_model_set_deployment_config_incompatible_instance_type_or_name(
         mock_get_model_specs.reset_mock()
         mock_model_deploy.reset_mock()
         mock_get_model_specs.side_effect = get_prototype_spec_with_configs
-        with pytest.raises(ValueError) as error:
-            model.set_deployment_config("neuron-inference", "ml.inf2.32xlarge")
-        assert (
-            "Instance type ml.inf2.32xlarge is not supported for config neuron-inference."
-            in str(error)
-        )
 
         with pytest.raises(ValueError) as error:
             model.set_deployment_config("neuron-inference-unknown-name", "ml.inf2.32xlarge")