fix tests

getmoto · Apr 22, 2024 · 4aed72c · 4aed72c
1 parent 4e445d7
commit 4aed72c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/moto/sagemaker/models.py b/moto/sagemaker/models.py
@@ -3795,7 +3795,7 @@ def gen_metrics_response_object(
             metrics_response_object = {
                 "MetricName": metrics_name,
                 "SourceArn": self.trial_component_arn,
-                "TimeStamp": datetime.fromtimestamp(timestamp_int).strftime(
+                "TimeStamp": datetime.fromtimestamp(timestamp_int, tz=tzutc()).strftime(
                     "%Y-%m-%d %H:%M:%S"
                 ),
                 "Max": max(metrics_steps_values),

diff --git a/tests/test_sagemakermetrics/test_sagemakermetrics.py b/tests/test_sagemakermetrics/test_sagemakermetrics.py
@@ -17,7 +17,7 @@ def test_batch_put_metrics():
     )
 
     client = boto3.client("sagemaker-metrics", region_name="eu-west-1")
-    given_datetime = datetime.datetime(2024, 4, 21, 19, 33, 3)
+    given_datetime = datetime.datetime(2024, 4, 21, 0, 0, 0)
     resp = client.batch_put_metrics(
         TrialComponentName=trial_component_name,
         MetricData=[
@@ -41,9 +41,7 @@ def test_batch_put_metrics():
         describe_after_metrics["Metrics"][0]["SourceArn"]
         == "arn:aws:sagemaker:eu-west-1:123456789012:experiment-trial-component/some-trial-component-name"
     )
-    assert describe_after_metrics["Metrics"][0]["TimeStamp"] == datetime.datetime(
-        2024, 4, 21, 21, 33, 3
-    )
+    assert describe_after_metrics["Metrics"][0]["TimeStamp"] == given_datetime
     assert describe_after_metrics["Metrics"][0]["Max"] == 123.0
     assert describe_after_metrics["Metrics"][0]["Min"] == 123.0
     assert describe_after_metrics["Metrics"][0]["Last"] == 123.0