Adding Hyperpod feature to enable hyperpod telemetry (#5235)

rsareddy0329 · Roja Reddy Sareddy · web-flow · commit 3641c2b9a149 · 2025-07-11T10:13:44.000-07:00
* Adding Hyperpod feature to enable hyperpod telemetry

* Adding Hyperpod feature to enable hyperpod telemetry

---------

Co-authored-by: Roja Reddy Sareddy &lt;rsareddy@amazon.com&gt;
diff --git a/src/sagemaker/telemetry/constants.py b/src/sagemaker/telemetry/constants.py
@@ -27,7 +27,7 @@ class Feature(Enum):
     REMOTE_FUNCTION = 3
     MODEL_TRAINER = 4
     ESTIMATOR = 5
-    HYPERPOD = 6
+    HYPERPOD = 6  # Added to support telemetry in sagemaker-hyperpod-cli
 
     def __str__(self):  # pylint: disable=E0307
         """Return the feature name."""
diff --git a/src/sagemaker/telemetry/telemetry_logging.py b/src/sagemaker/telemetry/telemetry_logging.py
@@ -55,7 +55,7 @@
     str(Feature.REMOTE_FUNCTION): 3,
     str(Feature.MODEL_TRAINER): 4,
     str(Feature.ESTIMATOR): 5,
-    str(Feature.HYPERPOD): 6,
+    str(Feature.HYPERPOD): 6,  # Added to support telemetry in sagemaker-hyperpod-cli
 }
 
 STATUS_TO_CODE = {

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@`
`55`	`55`	`str(Feature.REMOTE_FUNCTION): 3,`
`56`	`56`	`str(Feature.MODEL_TRAINER): 4,`
`57`	`57`	`str(Feature.ESTIMATOR): 5,`
`58`		`- str(Feature.HYPERPOD): 6,`
	`58`	`+ str(Feature.HYPERPOD): 6, # Added to support telemetry in sagemaker-hyperpod-cli`
`59`	`59`	`}`
`60`	`60`
`61`	`61`	`STATUS_TO_CODE = {`