diff --git a/sdk/python/feast/infra/offline_stores/snowflake.py b/sdk/python/feast/infra/offline_stores/snowflake.py index 14752fd8572..cfaca038e79 100644 --- 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 index b601c901462..0459cbd6eb2 100644 --- 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 = [