From 7cedc771d48db5d18048696a8083be59cc3e1437 Mon Sep 17 00:00:00 2001 From: Jared Duke Date: Wed, 27 May 2020 11:26:43 -0700 Subject: [PATCH] Internal testing change for coverage tests PiperOrigin-RevId: 313428290 Change-Id: I2a9d6684ddb26e71ea5be4de4c1ed7c1b11cb0d8 --- tensorflow/lite/build_def.bzl | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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",