From 7c66a41dc53165f47ac77fa139d6967cb81e2f94 Mon Sep 17 00:00:00 2001 From: pintaoz Date: Fri, 28 Feb 2025 11:03:30 -0800 Subject: [PATCH] Fix key error in _send_metrics() --- src/sagemaker/experiments/_metrics.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/experiments/_metrics.py b/src/sagemaker/experiments/_metrics.py index 31dd679cc8..026e73e8a6 100644 --- a/src/sagemaker/experiments/_metrics.py +++ b/src/sagemaker/experiments/_metrics.py @@ -197,8 +197,8 @@ def _send_metrics(self, metrics): response = self._metrics_client.batch_put_metrics(**request) errors = response["Errors"] if "Errors" in response else None if errors: - message = errors[0]["Message"] - raise Exception(f'{len(errors)} errors with message "{message}"') + error_code = errors[0]["Code"] + raise Exception(f'{len(errors)} errors with error code "{error_code}"') def _construct_batch_put_metrics_request(self, batch): """Creates dictionary object used as request to metrics service."""