From e8f03b0ebbea6eb3cb8a46f06d1b62cd7f9148b2 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sun, 27 Aug 2023 18:15:17 -0400 Subject: [PATCH] drop unnecessary REQUEST_ID column --- pyproject.toml | 2 +- wyvern/feature_store/feature_server.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 07a605d..6b2c092 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "wyvern-ai" -version = "0.0.12" +version = "0.0.13" description = "" authors = ["Wyvern AI "] readme = "README.md" diff --git a/wyvern/feature_store/feature_server.py b/wyvern/feature_store/feature_server.py index 828e79c..1da7257 100644 --- a/wyvern/feature_store/feature_server.py +++ b/wyvern/feature_store/feature_server.py @@ -366,6 +366,7 @@ async def get_historical_features( right_on=["REQUEST_ID", entity_identifier_type.upper()], how="left", ) + df.drop(columns=["REQUEST_ID"], inplace=True) feast_requests = build_historical_registry_feature_requests( store=store,