Added missing remote offline store apis implementation

Signed-off-by: Theodor Mihalache <[email protected]>
tmihalac · Jun 4, 2024 · cf38149 · cf38149
1 parent f3f44c3
commit cf38149
Show file tree

Hide file tree

Showing 7 changed files with 537 additions and 164 deletions.
diff --git a/sdk/python/feast/feature_store.py b/sdk/python/feast/feature_store.py
@@ -1121,6 +1121,11 @@ def create_saved_dataset(
         dataset.min_event_timestamp = from_.metadata.min_event_timestamp
         dataset.max_event_timestamp = from_.metadata.max_event_timestamp
 
+        # Added to handle the case that the offline store is remote
+        self._registry.apply_data_source(
+            storage.to_data_source(), self.project, commit=True
+        )
+
         from_.persist(storage=storage, allow_overwrite=allow_overwrite)
 
         dataset = dataset.with_retrieval_job(