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

j1wonpark · web-flow · commit c3dd74e238b5 · 2023-05-22T22:50:55.000-07:00
* fix: Fix scan datasize to 0 for inference schema

Signed-off-by: Jiwon Park &lt;bakjeeone@hotmail.com&gt;

* style: Rename queryRes to snake case

Signed-off-by: Jiwon Park &lt;bakjeeone@hotmail.com&gt;

---------

Signed-off-by: Jiwon Park &lt;bakjeeone@hotmail.com&gt;
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:
