From a957e1061717493af96326b4f4e0063695e69fed Mon Sep 17 00:00:00 2001 From: Danny Chiao Date: Wed, 10 Aug 2022 15:49:06 -0400 Subject: [PATCH] fix tests Signed-off-by: Danny Chiao --- sdk/python/feast/on_demand_feature_view.py | 3 ++- .../tests/integration/feature_repos/universal/feature_views.py | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/sdk/python/feast/on_demand_feature_view.py b/sdk/python/feast/on_demand_feature_view.py index 73ea451e89..fcafeaa2bc 100644 --- a/sdk/python/feast/on_demand_feature_view.py +++ b/sdk/python/feast/on_demand_feature_view.py @@ -83,7 +83,7 @@ def __init__( # noqa: C901 ] ], udf: FunctionType, - udf_string: str, + udf_string: str = "", description: str = "", tags: Optional[Dict[str, str]] = None, owner: str = "", @@ -140,6 +140,7 @@ def __copy__(self): sources=list(self.source_feature_view_projections.values()) + list(self.source_request_sources.values()), udf=self.udf, + udf_string=self.udf_string, description=self.description, tags=self.tags, owner=self.owner, diff --git a/sdk/python/tests/integration/feature_repos/universal/feature_views.py b/sdk/python/tests/integration/feature_repos/universal/feature_views.py index d7a0ac1df8..4eece13412 100644 --- a/sdk/python/tests/integration/feature_repos/universal/feature_views.py +++ b/sdk/python/tests/integration/feature_repos/universal/feature_views.py @@ -69,6 +69,7 @@ def conv_rate_plus_100_feature_view( schema=[] if infer_features else _features, sources=sources, udf=conv_rate_plus_100, + udf_string="raw udf source", ) @@ -106,6 +107,7 @@ def similarity_feature_view( sources=sources, schema=[] if infer_features else _fields, udf=similarity, + udf_string="similarity raw udf", )