Feature/js mlops telemetry

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")
@@ -1357,6 +1359,7 @@ def compile(
 
         return self
 
+    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="model.deploy")
     def deploy(
         self,
         initial_instance_count=None,