Fix

Author: nargokul

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

@@ -670,8 +670,6 @@ def prepare_fg_from_dataframe_or_file(
         raise exc
 
     if event_id not in data.columns:
-        import time
-
         current_time_sec = int(round(time.time()))
         data[event_id] = Series([current_time_sec] * lg_id, dtype="float64")
 

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

@@ -309,7 +309,7 @@ def test_with_latest_ingestion_and_event_time(self, mock_fg_class):
         mock_fg = MagicMock()
         mock_athena_query = MagicMock()
         mock_athena_query.table_name = "my_table"
-        mock_athena_query.as_dataframe.return_value = pd.DataFrame({"id": [1], "event_time": [123]})
+        mock_athena_query.as_dataframe.return_value = pd.DataFrame({"id": [1, 2], "event_time": [123, 123]})
         mock_fg.athena_query.return_value = mock_athena_query
         mock_fg_class.return_value = mock_fg