diff --git a/tensorflow/lite/build_def.bzl b/tensorflow/lite/build_def.bzl index f6cdb981328..fd51ad0a4aa 100644 --- a/tensorflow/lite/build_def.bzl +++ b/tensorflow/lite/build_def.bzl @@ -680,6 +680,9 @@ def gen_model_coverage_test(src, model_name, data, failure_type, tags, size = "m if failure_type[i] != "none": args.append("--failure_type=%s" % failure_type[i]) i = i + 1 + + # Avoid coverage timeouts for large/enormous tests. + coverage_tags = ["nozapfhahn"] if size in ["large", "enormous"] else [] native.py_test( name = "model_coverage_test_%s_%s" % (model_name, target_op_sets.lower().replace(",", "_")), srcs = [src], @@ -696,7 +699,7 @@ def gen_model_coverage_test(src, model_name, data, failure_type, tags, size = "m "no_gpu", # Executing with TF GPU configurations is redundant. "no_oss", "no_windows", - ] + tags, + ] + tags + coverage_tags, deps = [ "//tensorflow/lite/testing/model_coverage:model_coverage_lib", "//tensorflow/lite/python:lite",