From 703f262f6dcae3ae0ba1398275654231a296a65a Mon Sep 17 00:00:00 2001
From: Ryan Tanaka <rrtanaka@amazon.com>
Date: Sat, 28 Mar 2026 16:12:28 -0700
Subject: [PATCH] feat: add telemetry emitter to ScriptProcessor and
 FrameworkProcessor run methods

---
 sagemaker-core/src/sagemaker/core/processing.py               | 4 ++++
 sagemaker-core/src/sagemaker/core/telemetry/constants.py      | 1 +
 .../src/sagemaker/core/telemetry/telemetry_logging.py         | 1 +
 3 files changed, 6 insertions(+)

diff --git a/sagemaker-core/src/sagemaker/core/processing.py b/sagemaker-core/src/sagemaker/core/processing.py
index 736eebf01c..b507ae1a93 100644
--- a/sagemaker-core/src/sagemaker/core/processing.py
+++ b/sagemaker-core/src/sagemaker/core/processing.py
@@ -77,6 +77,8 @@
 from sagemaker.core.workflow.execution_variables import ExecutionVariables
 from sagemaker.core.workflow.functions import Join
 from sagemaker.core.workflow.pipeline_context import runnable_by_pipeline
+from sagemaker.core.telemetry.telemetry_logging import _telemetry_emitter
+from sagemaker.core.telemetry.constants import Feature
 
 from sagemaker.core._studio import _append_project_tags
 from sagemaker.core.config.config_utils import _append_sagemaker_config_tags
@@ -771,6 +773,7 @@ def __init__(
             network_config=network_config,
         )
 
+    @_telemetry_emitter(feature=Feature.PROCESSING, func_name="ScriptProcessor.run")
     @runnable_by_pipeline
     def run(
         self,
@@ -1171,6 +1174,7 @@ def _package_code(
             os.unlink(tmp.name)
             return s3_uri
 
+    @_telemetry_emitter(feature=Feature.PROCESSING, func_name="FrameworkProcessor.run")
     @runnable_by_pipeline
     def run(
         self,
diff --git a/sagemaker-core/src/sagemaker/core/telemetry/constants.py b/sagemaker-core/src/sagemaker/core/telemetry/constants.py
index 8b0bf9e5c3..2cd1fd44c0 100644
--- a/sagemaker-core/src/sagemaker/core/telemetry/constants.py
+++ b/sagemaker-core/src/sagemaker/core/telemetry/constants.py
@@ -29,6 +29,7 @@ class Feature(Enum):
     MODEL_CUSTOMIZATION = 15
     MLOPS = 16
     FEATURE_STORE = 17
+    PROCESSING = 18
 
     def __str__(self):  # pylint: disable=E0307
         """Return the feature name."""
diff --git a/sagemaker-core/src/sagemaker/core/telemetry/telemetry_logging.py b/sagemaker-core/src/sagemaker/core/telemetry/telemetry_logging.py
index ad54e95014..1041502064 100644
--- a/sagemaker-core/src/sagemaker/core/telemetry/telemetry_logging.py
+++ b/sagemaker-core/src/sagemaker/core/telemetry/telemetry_logging.py
@@ -61,6 +61,7 @@
     str(Feature.MODEL_CUSTOMIZATION): 15,
     str(Feature.MLOPS): 16,
     str(Feature.FEATURE_STORE): 17,
+    str(Feature.PROCESSING): 18,
 }
 
 STATUS_TO_CODE = {