chore: Increase snowflake-connector-python dependency to 3.7 (#4008)

* increase snowflake-connector-python to 3.7 Signed-off-by: Chester Ong <[email protected]> * force_return_table to True Signed-off-by: Chester Ong <[email protected]> --------- Signed-off-by: Chester Ong <[email protected]>
feast-dev · Mar 11, 2024 · 2076938 · 2076938
1 parent 924f944
commit 2076938
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 14 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/snowflake.py b/sdk/python/feast/infra/offline_stores/snowflake.py
@@ -470,20 +470,9 @@ def _to_df_internal(self, timeout: Optional[int] = None) -> pd.DataFrame:
         return df
 
     def _to_arrow_internal(self, timeout: Optional[int] = None) -> pyarrow.Table:
-        pa_table = execute_snowflake_statement(
+        return execute_snowflake_statement(
             self.snowflake_conn, self.to_sql()
-        ).fetch_arrow_all(force_return_table=False)
-
-        if pa_table:
-            return pa_table
-        else:
-            empty_result = execute_snowflake_statement(
-                self.snowflake_conn, self.to_sql()
-            )
-
-            return pyarrow.Table.from_pandas(
-                pd.DataFrame(columns=[md.name for md in empty_result.description])
-            )
+        ).fetch_arrow_all(force_return_table=True)
 
     def to_sql(self) -> str:
         """

diff --git a/setup.py b/setup.py
@@ -96,7 +96,7 @@
 BYTEWAX_REQUIRED = ["bytewax==0.15.1", "docker>=5.0.2", "kubernetes<=20.13.0"]
 
 SNOWFLAKE_REQUIRED = [
-    "snowflake-connector-python[pandas]>=3,<4",
+    "snowflake-connector-python[pandas]>=3.7,<4",
 ]
 
 SPARK_REQUIRED = [