Fix table_refs in BigQuerySource definitions (#1644)

* Fix table_refs in BigQuerySource definitions Signed-off-by: Tsotne Tabidze <tsotne@tecton.ai> * Replace with feast-oss Signed-off-by: Tsotne Tabidze <tsotne@tecton.ai> * Fix lint Signed-off-by: Tsotne Tabidze <tsotne@tecton.ai>
feast-dev · Jun 11, 2021 · eef6b98 · eef6b98
1 parent e63737c
commit eef6b98
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/sdk/python/tests/example_feature_repo_1.py b/sdk/python/tests/example_feature_repo_1.py
@@ -3,18 +3,17 @@
 from feast import BigQuerySource, Entity, Feature, FeatureView, ValueType
 
 driver_locations_source = BigQuerySource(
-    table_ref="rh_prod.ride_hailing_co.drivers",
+    table_ref="feast-oss.public.drivers",
     event_timestamp_column="event_timestamp",
     created_timestamp_column="created_timestamp",
 )
 
 customer_profile_source = BigQuerySource(
-    table_ref="rh_prod.ride_hailing_co.customers",
-    event_timestamp_column="event_timestamp",
+    table_ref="feast-oss.public.customers", event_timestamp_column="event_timestamp",
 )
 
 customer_driver_combined_source = BigQuerySource(
-    table_ref="rh_prod.ride_hailing_co.customer_driver",
+    table_ref="feast-oss.public.customer_driver",
     event_timestamp_column="event_timestamp",
 )
 

diff --git a/sdk/python/tests/test_partial_apply.py b/sdk/python/tests/test_partial_apply.py
@@ -17,7 +17,7 @@ def test_partial() -> None:
     ) as store:
 
         driver_locations_source = BigQuerySource(
-            table_ref="rh_prod.ride_hailing_co.drivers",
+            table_ref="feast-oss.public.drivers",
             event_timestamp_column="event_timestamp",
             created_timestamp_column="created_timestamp",
         )