diff --git a/aligned/feature_store.py b/aligned/feature_store.py index b4695da..aedcb26 100644 --- a/aligned/feature_store.py +++ b/aligned/feature_store.py @@ -470,7 +470,9 @@ class MyFeatureView: """ self.feature_views[view.name] = view if isinstance(self.feature_source, BatchFeatureSource): - self.feature_source.sources[FeatureLocation.feature_view(view.name).identifier] = view.source + self.feature_source.sources[FeatureLocation.feature_view(view.name).identifier] = ( + view.materialized_source or view.source + ) def add_feature_view(self, feature_view: FeatureView) -> None: self.add_compiled_view(feature_view.compile_instance()) diff --git a/pyproject.toml b/pyproject.toml index 4c4d69a..b30157c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "aligned" -version = "0.0.47" +version = "0.0.48" description = "A scalable feature store that makes it easy to align offline and online ML systems" authors = ["Mats E. Mollestad "] license = "Apache-2.0"