Skip to content

Commit d7ed025

Browse files
author
Roja Reddy Sareddy
committed
Adding Hyperpod feature to enable hyperpod telemetry
1 parent d408594 commit d7ed025

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

src/sagemaker/telemetry/constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ class Feature(Enum):
2727
REMOTE_FUNCTION = 3
2828
MODEL_TRAINER = 4
2929
ESTIMATOR = 5
30+
HYPERPOD = 6
3031

3132
def __str__(self): # pylint: disable=E0307
3233
"""Return the feature name."""

src/sagemaker/telemetry/telemetry_logging.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@
5555
str(Feature.REMOTE_FUNCTION): 3,
5656
str(Feature.MODEL_TRAINER): 4,
5757
str(Feature.ESTIMATOR): 5,
58+
str(Feature.HYPERPOD): 6,
5859
}
5960

6061
STATUS_TO_CODE = {

0 commit comments

Comments
 (0)