"tests/entrypoints/pooling/basic/test_encode.py" did not exist on "6f403501a085f4917e49e1714bdf44d2aabd06f9"
Unverified Commit c5932e5d authored by Eldar Kurtic's avatar Eldar Kurtic Committed by GitHub
Browse files

Properly check if all fused layers are in the list of targets (#12666)

Thanks @kylesayrs for catching this!
parent 20579c0f
...@@ -236,7 +236,7 @@ def _match_fused_layer(layer_name: str, ...@@ -236,7 +236,7 @@ def _match_fused_layer(layer_name: str,
for type_suffix in possible_layer_types) for type_suffix in possible_layer_types)
if is_same_parent and is_matching_type and all( if is_same_parent and is_matching_type and all(
'.'.join([parent_path, type_suffix]) (f"{parent_path}.{type_suffix}" in target_layers)
for type_suffix in possible_layer_types): for type_suffix in possible_layer_types):
return target return target
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment