From 337db3024e77cffc6ae74feec1692787baccca43 Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Fri, 29 Mar 2024 06:48:37 -0400 Subject: [PATCH] feat: update odfv test Signed-off-by: Francisco Javier Arceo --- sdk/python/tests/unit/test_on_demand_feature_view.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/tests/unit/test_on_demand_feature_view.py b/sdk/python/tests/unit/test_on_demand_feature_view.py index 02e013e7753..cf4afa94228 100644 --- a/sdk/python/tests/unit/test_on_demand_feature_view.py +++ b/sdk/python/tests/unit/test_on_demand_feature_view.py @@ -42,7 +42,7 @@ def udf2(features_df: pd.DataFrame) -> pd.DataFrame: return df -def python_native_udf(features_dict: Dict[str, List[Any]]) -> Dict[str, Any]: +def python_native_udf(features_dict: Dict[str, Any]) -> Dict[str, Any]: output_dict: Dict[str, List[Any]] = { "output1": features_dict["feature1"] + 100, "output2": features_dict["feature2"] + 101,