From 03bc968d67c368928c7c3a193e5eb167c3ef9b97 Mon Sep 17 00:00:00 2001
From: Jaesung Chung <jaesung@google.com>
Date: Mon, 13 Jul 2020 06:42:15 -0700
Subject: [PATCH] Add shard count to tflite_convert_test target

This is for removing flakiness of the test target.

PiperOrigin-RevId: 320945448
Change-Id: I06eb829ae2411e51202aad8231c8f87cdc194a55
---
 tensorflow/lite/python/BUILD | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow/lite/python/BUILD b/tensorflow/lite/python/BUILD
index 453aa803515..63be9602bc2 100644
--- a/tensorflow/lite/python/BUILD
+++ b/tensorflow/lite/python/BUILD
@@ -96,6 +96,8 @@ py_test(
         "@tflite_mobilenet_ssd_quant_protobuf//:tflite_graph.pb",
     ],
     python_version = "PY3",
+    # Increased thread count for reducing timeout failures.
+    shard_count = 4,
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",