test: fix test code for pandas compatability

Signed-off-by: Hai Nguyen <[email protected]>

sdk/python/tests/integration/offline_store/test_universal_historical_retrieval.py

@@ -340,6 +340,8 @@ def test_historical_features_with_entities_from_query(
 
     table_from_sql_entities = job_from_sql.to_arrow().to_pandas()
     for col in table_from_sql_entities.columns:
+        if expected_df_query[col].dtype == "datetime64[ns]":
+            expected_df_query[col] = expected_df_query[col].dt.tz_localize("UTC")
         expected_df_query[col] = expected_df_query[col].astype(
             table_from_sql_entities[col].dtype
         )