|
21 | 21 |
|
22 | 22 | from feast.entity import Entity |
23 | 23 | from feast.feature_view import FeatureView |
24 | | -from feast.infra.provider import Provider, get_provider |
25 | | -from feast.offline_store import ( |
26 | | - RetrievalJob, |
27 | | - get_offline_store, |
28 | | - get_offline_store_for_retrieval, |
29 | | -) |
| 24 | +from feast.infra.provider import Provider, RetrievalJob, get_provider |
30 | 25 | from feast.online_response import OnlineResponse, _infer_online_entity_rows |
31 | 26 | from feast.protos.feast.serving.ServingService_pb2 import ( |
32 | 27 | GetOnlineFeaturesRequestV2, |
@@ -255,8 +250,8 @@ def get_historical_features( |
255 | 250 | project=self.config.project |
256 | 251 | ) |
257 | 252 | feature_views = _get_requested_feature_views(feature_refs, all_feature_views) |
258 | | - offline_store = get_offline_store_for_retrieval(feature_views) |
259 | | - job = offline_store.get_historical_features( |
| 253 | + provider = self._get_provider() |
| 254 | + job = provider.get_historical_features( |
260 | 255 | self.config, feature_views, feature_refs, entity_df |
261 | 256 | ) |
262 | 257 | return job |
@@ -367,9 +362,8 @@ def _materialize_single_feature_view( |
367 | 362 | event_timestamp_column, |
368 | 363 | created_timestamp_column, |
369 | 364 | ) = _run_reverse_field_mapping(feature_view) |
370 | | - |
371 | | - offline_store = get_offline_store(self.config) |
372 | | - table = offline_store.pull_latest_from_table_or_query( |
| 365 | + provider = self._get_provider() |
| 366 | + table = provider.pull_latest_from_table_or_query( |
373 | 367 | feature_view.input, |
374 | 368 | entity_names, |
375 | 369 | feature_names, |
|
0 commit comments