From 8ff0969b0ff672c237f396cb6c71de63833accc5 Mon Sep 17 00:00:00 2001
From: "Mats E. Mollestad" <mast@mollestad.no>
Date: Fri, 1 Dec 2023 14:54:16 +0100
Subject: [PATCH] Release New Version

---
 aligned/feature_store.py | 4 +++-
 pyproject.toml           | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

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 <mats@mollestad.no>"]
 license = "Apache-2.0"