[CONFORMANCE] Partial cherry-pick of PR20756 to exclude from model list for conformance (#20955)

This commit is contained in:
Irina Efode 2023-11-08 15:20:39 +04:00 committed by GitHub
parent 5ee4090e10
commit fc678416a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -90,6 +90,9 @@ def generate_model_list_file(input_str: str, re_exp_file_path: str, output_file_
except:
pass
for line in model_list:
str_line = str(line)
if "tfhub_module.pb" in str_line or "_metadata.pb" in str_line:
continue
output_file.write(f"{line}\n")
output_file.close()