From 74510f1d55544669253204879619b95001077e3f Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Thu, 14 Sep 2023 22:32:30 -0700
Subject: [PATCH] make feature store features available in realtime feature
 computation

---
 wyvern/components/features/feature_retrieval_pipeline.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/wyvern/components/features/feature_retrieval_pipeline.py b/wyvern/components/features/feature_retrieval_pipeline.py
index e8a073d..106f66c 100644
--- a/wyvern/components/features/feature_retrieval_pipeline.py
+++ b/wyvern/components/features/feature_retrieval_pipeline.py
@@ -6,6 +6,7 @@
 from ddtrace import tracer
 from pydantic.generics import GenericModel
 
+from wyvern import request_context
 from wyvern.components.component import Component
 from wyvern.components.features.feature_logger import (
     FeatureEventLoggingComponent,
@@ -156,6 +157,8 @@ async def execute(
                 **kwargs,
             )
         )
+        current_request = request_context.ensure_current_request()
+        current_request.feature_map = feature_retrieval_response
 
         """
         TODO (suchintan):