fix: Fix scan datasize to 0 for inference schema (feast-dev#3628)

* fix: Fix scan datasize to 0 for inference schema Signed-off-by: Jiwon Park <[email protected]> * style: Rename queryRes to snake case Signed-off-by: Jiwon Park <[email protected]> --------- Signed-off-by: Jiwon Park <[email protected]> Signed-off-by: zerafachris PERSONAL <[email protected]>
zerafachris · Mar 5, 2024 · 206c6a6 · 206c6a6
1 parent 721d3ba
commit 206c6a6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/bigquery_source.py b/sdk/python/feast/infra/offline_stores/bigquery_source.py
@@ -180,9 +180,9 @@ def get_table_column_names_and_types(
             if not isinstance(schema[0], bigquery.schema.SchemaField):
                 raise TypeError("Could not parse BigQuery table schema.")
         else:
-            bq_columns_query = f"SELECT * FROM ({self.query}) LIMIT 1"
-            queryRes = client.query(bq_columns_query).result()
-            schema = queryRes.schema
+            bq_columns_query = f"SELECT * FROM ({self.query}) LIMIT 0"
+            query_res = client.query(bq_columns_query).result()
+            schema = query_res.schema
 
         name_type_pairs: List[Tuple[str, str]] = []
         for field in schema: