From 4608b40a304a0b1f6e9692d43f4714b50e3222f2 Mon Sep 17 00:00:00 2001
From: Scott Zhu <scottzhu@google.com>
Date: Fri, 15 Nov 2019 11:59:57 -0800
Subject: [PATCH] Change the test size for data_adapter to reduce the timeout.

The base tests will take 40 seconds-ish to run, and sometimes its exceeding the small tests limit (60 seconds).

PiperOrigin-RevId: 280706537
Change-Id: I9e6fc5e7f50cdc392fbc08fa7cc9049630ef0cee
---
 tensorflow/python/keras/BUILD | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD
index 52ed1cbda56..0584d1ca97f 100755
--- a/tensorflow/python/keras/BUILD
+++ b/tensorflow/python/keras/BUILD
@@ -617,7 +617,7 @@ tf_py_test(
 
 tf_py_test(
     name = "data_adapter_test",
-    size = "small",
+    size = "medium",
     srcs = ["engine/data_adapter_test.py"],
     additional_deps = [
         ":data_adapter",