chore: Update skip reason

sudohainguyen · sudohainguyen · commit 2f83b8bf949c · 2024-02-19T13:43:57.000+07:00
Signed-off-by: Hai Nguyen &lt;quanghai.ng1512@gmail.com&gt;
diff --git a/sdk/python/tests/integration/offline_store/test_feature_logging.py b/sdk/python/tests/integration/offline_store/test_feature_logging.py
@@ -30,7 +30,7 @@
 @pytest.mark.parametrize("pass_as_path", [True, False], ids=lambda v: str(v))
 @pytest.mark.skipif(
     pd.__version__ < "2.2.0" and pd.__version__ >= "2.0.0",
-    reason="Requires pandas version 2.2.0 or higher",
+    reason="Skip test due to pandas issue 55730 for pandas version 2.0.0 - 2.1.0",
     # https://github.com/pandas-dev/pandas/issues/55730
 )
 def test_feature_service_logging(environment, universal_data_sources, pass_as_path):
diff --git a/sdk/python/tests/integration/offline_store/test_universal_historical_retrieval.py b/sdk/python/tests/integration/offline_store/test_universal_historical_retrieval.py
@@ -340,6 +340,11 @@ 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:
+        # check if col dtype is timezone naive
+        if pd.api.types.is_datetime64_dtype(table_from_sql_entities[col]):
+            table_from_sql_entities[col] = table_from_sql_entities[col].dt.tz_localize(
+                "UTC"
+            )
         expected_df_query[col] = expected_df_query[col].astype(
             table_from_sql_entities[col].dtype
         )
@@ -515,7 +520,7 @@ def test_historical_features_with_no_ttl(
 @pytest.mark.universal_offline_stores
 @pytest.mark.skipif(
     pd.__version__ < "2.2.0" and pd.__version__ >= "2.0.0",
-    reason="Requires pandas version 2.2.0 or higher",
+    reason="Skip test due to pandas issue 55730 for pandas version 2.0.0 - 2.1.0",
     # https://github.com/pandas-dev/pandas/issues/55730
 )
 def test_historical_features_from_bigquery_sources_containing_backfills(environment):

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@`
`30`	`30`	`@pytest.mark.parametrize("pass_as_path", [True, False], ids=lambda v: str(v))`
`31`	`31`	`@pytest.mark.skipif(`
`32`	`32`	`pd.__version__ < "2.2.0" and pd.__version__ >= "2.0.0",`
`33`		`- reason="Requires pandas version 2.2.0 or higher",`
	`33`	`+ reason="Skip test due to pandas issue 55730 for pandas version 2.0.0 - 2.1.0",`
`34`	`34`	`# https://github.com/pandas-dev/pandas/issues/55730`
`35`	`35`	`)`
`36`	`36`	`def test_feature_service_logging(environment, universal_data_sources, pass_as_path):`