reverting feature store change

Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
feast-dev · Jan 29, 2025 · 2a02f2f · 2a02f2f
1 parent ed10ab4
commit 2a02f2f
Showing 1 changed file with 13 additions and 24 deletions.
diff --git a/sdk/python/feast/feature_store.py b/sdk/python/feast/feature_store.py
@@ -1847,30 +1847,19 @@ def retrieve_online_documents(
                         join_key_values[join_key] = []
                     join_key_values[join_key].append(entity_value)
 
-            document_feature_vals = [feature[4] for feature in document_features]
-            document_feature_distance_vals = [
-                feature[5] for feature in document_features
-            ]
-            online_features_response = GetOnlineFeaturesResponse(results=[])
-            requested_feature = requested_feature or requested_features[0]
-            utils._populate_result_rows_from_columnar(
-                online_features_response=online_features_response,
-                data={
-                    **join_key_values,
-                    requested_feature: document_feature_vals,
-                    "distance": document_feature_distance_vals,
-                },
-            )
-            return OnlineResponse(online_features_response)
-        else:
-            return self._retrieve_from_online_store_v2(
-                provider,
-                requested_feature_view,
-                requested_features,
-                query,
-                top_k,
-                distance_metric,
-            )
+        document_feature_vals = [feature[4] for feature in document_features]
+        document_feature_distance_vals = [feature[5] for feature in document_features]
+        online_features_response = GetOnlineFeaturesResponse(results=[])
+        requested_feature = requested_feature or requested_features[0]
+        utils._populate_result_rows_from_columnar(
+            online_features_response=online_features_response,
+            data={
+                **join_key_values,
+                requested_feature: document_feature_vals,
+                "distance": document_feature_distance_vals,
+            },
+        )
+        return OnlineResponse(online_features_response)
 
     def retrieve_online_documents_v2(
         self,