Skip to content

Commit c2426fe

Browse files
committed
Fixed event_time format in integ tests
1 parent bdcbcf7 commit c2426fe

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

sagemaker-mlops/tests/integ/test_feature_store_iceberg_properties.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import time
33

44
import boto3
5+
56
import pandas as pd
67
import pytest
78

@@ -45,12 +46,13 @@ def feature_group_name():
4546

4647
@pytest.fixture
4748
def sample_dataframe():
48-
current_time = int(time.time())
49+
from datetime import datetime, timezone, timedelta
50+
base_time = datetime.now(timezone.utc)
4951
return pd.DataFrame({
5052
"record_id": [f"id-{i}" for i in range(10)],
5153
"feature_1": [i * 1.5 for i in range(10)],
5254
"feature_2": [i * 2 for i in range(10)],
53-
"event_time": [float(current_time + i) for i in range(10)],
55+
"event_time": [(base_time + timedelta(seconds=i)).strftime("%Y-%m-%dT%H:%M:%SZ") for i in range(10)],
5456
})
5557

5658

0 commit comments

Comments
 (0)