feature: Add Mariadb offline store

Added MariaDB template Signed-off-by: Theodor Mihalache <[email protected]>
tmihalac · May 3, 2024 · 075b0d7 · 075b0d7
1 parent 0c303ff
commit 075b0d7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/contrib/mariadb_offline_store/mariadb.py b/sdk/python/feast/infra/offline_stores/contrib/mariadb_offline_store/mariadb.py
@@ -217,7 +217,9 @@ def get_historical_features(
             engine=engine,
             config=config.offline_store,
             full_feature_names=full_feature_names,
-            on_demand_feature_views=OnDemandFeatureView.get_requested_odfvs(feature_refs, project, registry),
+            on_demand_feature_views=OnDemandFeatureView.get_requested_odfvs(
+                feature_refs, project, registry
+            ),
         )
         return job
 
@@ -234,10 +236,10 @@ def write_logged_features(
     @staticmethod
     @log_exceptions_and_usage(offline_store="mariadb")
     def offline_write_batch(
-            config: RepoConfig,
-            feature_view: FeatureView,
-            table: pyarrow.Table,
-            progress: Optional[Callable[[int], Any]],
+        config: RepoConfig,
+        feature_view: FeatureView,
+        table: pyarrow.Table,
+        progress: Optional[Callable[[int], Any]],
     ):
         assert isinstance(config.offline_store, MariaDBOfflineStoreConfig)
         assert type(feature_view.batch_source).__name__ == "MariaDBSource"

diff --git a/sdk/python/feast/infra/offline_stores/contrib/mssql_offline_store/mssql.py b/sdk/python/feast/infra/offline_stores/contrib/mssql_offline_store/mssql.py
@@ -233,7 +233,9 @@ def get_historical_features(
             engine=engine,
             config=config.offline_store,
             full_feature_names=full_feature_names,
-            on_demand_feature_views=OnDemandFeatureView.get_requested_odfvs(feature_refs, project, registry),
+            on_demand_feature_views=OnDemandFeatureView.get_requested_odfvs(
+                feature_refs, project, registry
+            ),
         )
         return job