chore: improve spec field name

Author: evakravi

src/sagemaker/jumpstart/types.py

@@ -730,7 +730,7 @@ class JumpStartModelSpecs(JumpStartDataHolderType):
         "training_dependencies",
         "training_vulnerabilities",
         "deprecated",
-        "info_message",
+        "usage_info_message",
         "deprecated_message",
         "deprecate_warn_message",
         "default_inference_instance_type",
@@ -802,7 +802,7 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:
         self.deprecated: bool = bool(json_obj["deprecated"])
         self.deprecated_message: Optional[str] = json_obj.get("deprecated_message")
         self.deprecate_warn_message: Optional[str] = json_obj.get("deprecate_warn_message")
-        self.info_message: Optional[str] = json_obj.get("info_message")
+        self.usage_info_message: Optional[str] = json_obj.get("usage_info_message")
         self.default_inference_instance_type: Optional[str] = json_obj.get(
             "default_inference_instance_type"
         )

src/sagemaker/jumpstart/utils.py

@@ -511,8 +511,8 @@ def emit_logs_based_on_model_specs(
     if model_specs.deprecate_warn_message:
         constants.JUMPSTART_LOGGER.warning(model_specs.deprecate_warn_message)
 
-    if model_specs.info_message:
-        constants.JUMPSTART_LOGGER.info(model_specs.info_message)
+    if model_specs.usage_info_message:
+        constants.JUMPSTART_LOGGER.info(model_specs.usage_info_message)
 
     if model_specs.inference_vulnerable or model_specs.training_vulnerable:
         constants.JUMPSTART_LOGGER.warning(

tests/unit/sagemaker/jumpstart/constants.py

@@ -6221,7 +6221,7 @@
         "ml.c5.2xlarge",
     ],
     "hosting_use_script_uri": True,
-    "info_message": None,
+    "usage_info_message": None,
     "metrics": [{"Regex": "val_accuracy: ([0-9\\.]+)", "Name": "pytorch-ic:val-accuracy"}],
     "model_kwargs": {"some-model-kwarg-key": "some-model-kwarg-value"},
     "deploy_kwargs": {"some-model-deploy-kwarg-key": "some-model-deploy-kwarg-value"},

tests/unit/sagemaker/jumpstart/test_utils.py

@@ -1034,20 +1034,20 @@ def make_deprecated_warning_message_spec(*largs, **kwargs):
 
 
 @patch("sagemaker.jumpstart.utils.accessors.JumpStartModelsAccessor._get_manifest")
-def test_jumpstart_info_message(mock_get_manifest):
+def test_jumpstart_usage_info_message(mock_get_manifest):
     mock_get_manifest.return_value = []
 
-    info_message = "This model might change your life."
+    usage_info_message = "This model might change your life."
 
     def make_info_spec(*largs, **kwargs):
         spec = get_spec_from_base_spec(model_id="pytorch-eqa-bert-base-cased", version="*")
-        spec.info_message = info_message
+        spec.usage_info_message = usage_info_message
         return spec
 
     with patch("logging.Logger.info") as mocked_info_log:
         utils.emit_logs_based_on_model_specs(make_info_spec(), "us-west-2", MOCK_CLIENT)
 
-        mocked_info_log.assert_called_with(info_message)
+        mocked_info_log.assert_called_with(usage_info_message)
 
 
 @patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor.get_model_specs")