fix: Method _should_use_plan only returns true for local sqlite provi…

…der (feast-dev#2344) Only the SQLite local provider supports _plan and _apply_diffs. Updated the method to only return true for that setup so that other local providers don't need to implement these methods right now. Signed-off-by: Gunnar Sv Sigurbjörnsson <[email protected]>
achals · Mar 1, 2022 · fdb5f21 · fdb5f21
1 parent 43cc886
commit fdb5f21
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdk/python/feast/feature_store.py b/sdk/python/feast/feature_store.py
@@ -392,8 +392,10 @@ def _get_features(
 
     def _should_use_plan(self):
         """Returns True if _plan and _apply_diffs should be used, False otherwise."""
-        # Currently only the local provider supports _plan and _apply_diffs.
-        return self.config.provider == "local"
+        # Currently only the local provider with sqlite online store supports _plan and _apply_diffs.
+        return self.config.provider == "local" and (
+            self.config.online_store and self.config.online_store.type == "sqlite"
+        )
 
     def _validate_all_feature_views(
         self,