diff --git a/src/sagemaker/telemetry/constants.py b/src/sagemaker/telemetry/constants.py index 2c803f4aa2..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 + 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..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, + str(Feature.HYPERPOD): 6, # Added to support telemetry in sagemaker-hyperpod-cli } STATUS_TO_CODE = {