feat: Update odfv test (#4054)

feat: update odfv test Signed-off-by: Francisco Javier Arceo <franciscojavierarceo@users.noreply.github.com>
feast-dev · Mar 29, 2024 · afd52b8 · afd52b8
1 parent df05253
commit afd52b8
Showing 1 changed file with 1 addition and 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
@@ -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,