Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
lint
Signed-off-by: Francisco Javier Arceo <franciscojavierarceo@users.noreply.github.com>
  • Loading branch information
franciscojavierarceo committed Apr 4, 2024
commit 1d3745d6f0d1833295187db15763685b70b8cb1c
16 changes: 8 additions & 8 deletions sdk/python/tests/unit/test_on_demand_feature_view.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,14 +185,14 @@ def test_python_native_transformation_mode():
)

assert (
on_demand_feature_view_python_native.feature_transformation
== PythonTransformation(python_native_udf, "python native udf source code")
on_demand_feature_view_python_native.feature_transformation
== PythonTransformation(python_native_udf, "python native udf source code")
)

with pytest.raises(TypeError):
assert (
on_demand_feature_view_python_native_err.feature_transformation
== PythonTransformation(python_native_udf, "python native udf source code")
on_demand_feature_view_python_native_err.feature_transformation
== PythonTransformation(python_native_udf, "python native udf source code")
)

with pytest.raises(TypeError):
Expand Down Expand Up @@ -241,8 +241,8 @@ def test_from_proto_backwards_compatible_udf():
# and to populate it in feature_transformation
proto = on_demand_feature_view.to_proto()
assert (
on_demand_feature_view.feature_transformation.udf_string
== proto.spec.feature_transformation.user_defined_function.body_text
on_demand_feature_view.feature_transformation.udf_string
== proto.spec.feature_transformation.user_defined_function.body_text
)
# Because of the current set of code this is just confirming it is empty
assert proto.spec.user_defined_function.body_text == ""
Expand All @@ -269,6 +269,6 @@ def test_from_proto_backwards_compatible_udf():
# And now we expect the to get the same object back under feature_transformation
reserialized_proto = OnDemandFeatureView.from_proto(proto)
assert (
reserialized_proto.feature_transformation.udf_string
== on_demand_feature_view.feature_transformation.udf_string
reserialized_proto.feature_transformation.udf_string
== on_demand_feature_view.feature_transformation.udf_string
)