fix

Author: nargokul

sagemaker-mlops/src/sagemaker/mlops/feature_store/feature_utils.py

@@ -11,7 +11,7 @@
 import boto3
 import pandas
 import pandas as pd
-from pandas import DataFrame, Series, read_csv
+from pandas import DataFrame, Series, read_csv\
 
 from sagemaker.mlops.feature_store import FeatureGroup as CoreFeatureGroup, FeatureGroup
 from sagemaker.core.helper.session_helper import Session
@@ -670,7 +670,7 @@ def prepare_fg_from_dataframe_or_file(
         raise exc
 
     if event_id not in data.columns:
-        current_time_sec = int(round(time.time()))
+        current_time_sec = int(time.time() + 0.5)
         data[event_id] = Series([current_time_sec] * lg_id, dtype="float64")
 
     if session is not None:

sagemaker-mlops/tests/unit/sagemaker/mlops/feature_store/test_feature_utils.py

@@ -591,7 +591,7 @@ def test_creates_event_id_with_timestamp(self, mock_time, mock_fg_class):
         )
 
         assert "data_as_of_date" in df.columns
-        assert all(df["data_as_of_date"] == 1234567891)
+        assert all(df["data_as_of_date"] == 1234567891.0)
 
     @patch("sagemaker.mlops.feature_store.feature_utils.FeatureGroup")
     def test_uses_existing_event_id(self, mock_fg_class):