added telemetry for hub create, hub create model reference, model_package deploy

Author: Mohamed Zeidan

src/sagemaker/jumpstart/hub/hub.py

@@ -43,6 +43,8 @@
     LATEST_VERSION_WILDCARD,
 )
 from sagemaker.jumpstart import utils
+from sagemaker.telemetry.telemetry_logging import _telemetry_emitter
+from sagemaker.telemetry.constants import Feature
 
 
 class Hub:
@@ -84,6 +86,7 @@ def _get_latest_model_version(self, model_id: str) -> str:
         )
         return model_specs.version
 
+    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="hub.create")
     def create(
         self,
         description: str,
@@ -213,6 +216,7 @@ def delete(self) -> None:
         """Deletes this SageMaker Hub."""
         return self._sagemaker_session.delete_hub(self.hub_name)
 
+    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="hub.create_model_reference")
     def create_model_reference(
         self, model_arn: str, model_name: Optional[str] = None, min_version: Optional[str] = None
     ):

src/sagemaker/model.py

@@ -90,6 +90,8 @@
     get_update_model_package_inference_args,
 )
 from sagemaker.model_life_cycle import ModelLifeCycle
+from sagemaker.telemetry.telemetry_logging import _telemetry_emitter
+from sagemaker.telemetry.constants import Feature
 
 # Setting LOGGER for backward compatibility, in case users import it...
 logger = LOGGER = logging.getLogger("sagemaker")
@@ -2542,6 +2544,22 @@ def add_inference_specification(
 
         sagemaker_session.sagemaker_client.update_model_package(**model_package_update_args)
 
+    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="model_package.deploy")
+    def deploy(self, *args, **kwargs):
+        """Deploy this ``ModelPackage`` to an ``Endpoint`` and optionally return a ``Predictor``.
+
+        This method overrides the parent Model.deploy() method to add telemetry tracking
+        specifically for ModelPackage deployments, which are used by JumpStart proprietary models.
+
+        Args:
+            *args: Positional arguments passed to the parent deploy method.
+            **kwargs: Keyword arguments passed to the parent deploy method.
+
+        Returns:
+            The result of the parent deploy method.
+        """
+        return super().deploy(*args, **kwargs)
+
     def update_model_card(self, model_card: Union[ModelCard, ModelPackageModelCard]):
         """Updates Created model card content which created with model package