From d3294fa6169c572e5198a7b62a543d9193282667 Mon Sep 17 00:00:00 2001 From: Roja Reddy Sareddy Date: Fri, 11 Jul 2025 01:08:25 -0700 Subject: [PATCH 1/2] Adding Hyperpod feature to enable hyperpod telemetry --- src/sagemaker/telemetry/constants.py | 2 +- src/sagemaker/telemetry/telemetry_logging.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/telemetry/constants.py b/src/sagemaker/telemetry/constants.py index 2c803f4aa2..216c079f4a 100644 --- 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 index 8c431b3338..3444019614 100644 --- 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 = { From a583ecdd804fd91637561510b6a33ec5f803d657 Mon Sep 17 00:00:00 2001 From: Roja Reddy Sareddy Date: Fri, 11 Jul 2025 01:35:09 -0700 Subject: [PATCH 2/2] Adding Hyperpod feature to enable hyperpod telemetry --- src/sagemaker/telemetry/constants.py | 2 +- src/sagemaker/telemetry/telemetry_logging.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/telemetry/constants.py b/src/sagemaker/telemetry/constants.py index 216c079f4a..6766d45b4e 100644 --- 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 # Added to support telemetry in sagemaker-hyperpod-cli + 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 index 3444019614..990e12124f 100644 --- 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, # Added to support telemetry in sagemaker-hyperpod-cli + str(Feature.HYPERPOD): 6, # Added to support telemetry in sagemaker-hyperpod-cli } STATUS_TO_CODE = {