diff --git a/sdk/python/feast/feature_store.py b/sdk/python/feast/feature_store.py index 8c93a397673..d6409b25ec2 100644 --- a/sdk/python/feast/feature_store.py +++ b/sdk/python/feast/feature_store.py @@ -76,6 +76,7 @@ class FeatureStore: config: RepoConfig repo_path: Path _registry: Registry + _provider: Provider @log_exceptions def __init__( @@ -100,6 +101,7 @@ def __init__( registry_config = self.config.get_registry_config() self._registry = Registry(registry_config, repo_path=self.repo_path) + self._provider = get_provider(self.config, self.repo_path) @log_exceptions def version(self) -> str: @@ -118,7 +120,7 @@ def project(self) -> str: def _get_provider(self) -> Provider: # TODO: Bake self.repo_path into self.config so that we dont only have one interface to paths - return get_provider(self.config, self.repo_path) + return self._provider @log_exceptions_and_usage def refresh_registry(self):