Skip to content

Commit

Permalink
fix: fix test_tensorflow_deserialize by resolving the lambda filterin…
Browse files Browse the repository at this point in the history
…g for activation functions (#28689)
  • Loading branch information
ZJay07 committed Mar 28, 2024
1 parent 5173e82 commit eac08ee
Showing 1 changed file with 3 additions and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -89,17 +89,7 @@ def simple_test_two_function(
@handle_frontend_test(
fn_tree="tensorflow.keras.activations.deserialize",
fn_name=st.sampled_from(get_callable_functions("keras.activations")).filter(
lambda x: not x[0].isupper()
and x
not in [
"deserialize",
"get",
"keras_export",
"serialize",
"deserialize_keras_object",
"serialize_keras_object",
"get_globals",
]
lambda x: not x[0].isupper() and x in ACTIVATION_FUNCTIONS
),
dtype_and_data=helpers.dtype_and_values(
available_dtypes=helpers.get_dtypes("valid"),
Expand All @@ -114,6 +104,7 @@ def test_tensorflow_deserialize(
fn_name,
fn_tree,
frontend,
backend_fw,
):
dtype_data, data = dtype_and_data
simple_test_two_function(
Expand All @@ -126,6 +117,7 @@ def test_tensorflow_deserialize(
atol_=1e-01,
ivy_submodules=["keras", "activations"],
framework_submodules=["keras", "activations"],
backend=backend_fw,
)


Expand Down

0 comments on commit eac08ee

Please sign in to comment.