fix bigquery source

Signed-off-by: Danny Chiao <danny@tecton.ai>
feast-dev · Mar 2, 2022 · aad4b77 · aad4b77
1 parent d35fd9a
commit aad4b77
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/bigquery_source.py b/sdk/python/feast/infra/offline_stores/bigquery_source.py
@@ -44,14 +44,15 @@ def __init__(
          """
         if table is None and table_ref is None and query is None:
             raise ValueError('No "table" argument provided.')
-        if table_ref:
+        if not table and table_ref:
             warnings.warn(
                 (
                     "The argument 'table_ref' is being deprecated. Please use 'table' "
                     "instead. Feast 0.20 and onwards will not support the argument 'table_ref'."
                 ),
                 DeprecationWarning,
             )
+            table = table_ref
         self.bigquery_options = BigQueryOptions(table_ref=table, query=query)
 
         # If no name, use the table_ref as the default name

diff --git a/sdk/python/tests/example_repos/example_feature_repo_1.py b/sdk/python/tests/example_repos/example_feature_repo_1.py
@@ -10,7 +10,7 @@
 )
 
 driver_locations_source = BigQuerySource(
-    table_ref="feast-oss.public.drivers",
+    table="feast-oss.public.drivers",
     event_timestamp_column="event_timestamp",
     created_timestamp_column="created_timestamp",
 )