chore: address PR comment

Author: evakravi

src/sagemaker/jumpstart/utils.py

@@ -402,7 +402,7 @@ def verify_model_region_and_return_specs(
             f"JumpStart model ID '{model_id}' and version '{version}' " "does not support training."
         )
 
-    if model_specs.deprecated or model_specs.deprecated_message:
+    if model_specs.deprecated:
         if not tolerate_deprecated_model:
             raise DeprecatedJumpStartModelError(
                 model_id=model_id, version=version, message=model_specs.deprecated_message

tests/unit/sagemaker/jumpstart/test_utils.py

@@ -892,6 +892,7 @@ def make_deprecated_spec(*largs, **kwargs):
     def make_deprecated_message_spec(*largs, **kwargs):
         spec = get_spec_from_base_spec(*largs, **kwargs)
         spec.deprecated_message = deprecated_message
+        spec.deprecated = True
         return spec
 
     patched_get_model_specs.side_effect = make_deprecated_message_spec