File tree Expand file tree Collapse file tree 3 files changed +6
-0
lines changed
sagemaker-core/src/sagemaker/core Expand file tree Collapse file tree 3 files changed +6
-0
lines changed Original file line number Diff line number Diff line change 7777from sagemaker .core .workflow .execution_variables import ExecutionVariables
7878from sagemaker .core .workflow .functions import Join
7979from sagemaker .core .workflow .pipeline_context import runnable_by_pipeline
80+ from sagemaker .core .telemetry .telemetry_logging import _telemetry_emitter
81+ from sagemaker .core .telemetry .constants import Feature
8082
8183from sagemaker .core ._studio import _append_project_tags
8284from sagemaker .core .config .config_utils import _append_sagemaker_config_tags
@@ -771,6 +773,7 @@ def __init__(
771773 network_config = network_config ,
772774 )
773775
776+ @_telemetry_emitter (feature = Feature .PROCESSING , func_name = "ScriptProcessor.run" )
774777 @runnable_by_pipeline
775778 def run (
776779 self ,
@@ -1171,6 +1174,7 @@ def _package_code(
11711174 os .unlink (tmp .name )
11721175 return s3_uri
11731176
1177+ @_telemetry_emitter (feature = Feature .PROCESSING , func_name = "FrameworkProcessor.run" )
11741178 @runnable_by_pipeline
11751179 def run (
11761180 self ,
Original file line number Diff line number Diff line change @@ -29,6 +29,7 @@ class Feature(Enum):
2929 MODEL_CUSTOMIZATION = 15
3030 MLOPS = 16
3131 FEATURE_STORE = 17
32+ PROCESSING = 18
3233
3334 def __str__ (self ): # pylint: disable=E0307
3435 """Return the feature name."""
Original file line number Diff line number Diff line change 6161 str (Feature .MODEL_CUSTOMIZATION ): 15 ,
6262 str (Feature .MLOPS ): 16 ,
6363 str (Feature .FEATURE_STORE ): 17 ,
64+ str (Feature .PROCESSING ): 18 ,
6465}
6566
6667STATUS_TO_CODE = {
You can’t perform that action at this time.
0 commit comments