diff --git a/src/sagemaker/jumpstart/hub/hub.py b/src/sagemaker/jumpstart/hub/hub.py index 692966cee4..f990f2e5af 100644 --- a/src/sagemaker/jumpstart/hub/hub.py +++ b/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 ): diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py index 3bfac0c8da..c94e0ff29d 100644 --- a/src/sagemaker/model.py +++ b/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,