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 index b3c885b92a4..a905f4c200a 100644 --- 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 index 0cf292b6cb4..9dd80e29b73 100644 --- 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