Increase size of test tensorflow/python:basic_session_run_hooks_test to avoid flaky...

PiperOrigin-RevId: 196016436

diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index 699f78e..f7cbaec 100644 (file)
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -4219,7 +4219,7 @@ tf_py_test(
 
 py_test(
     name = "basic_session_run_hooks_test",
-    size = "small",
+    size = "medium",
     srcs = ["training/basic_session_run_hooks_test.py"],
     srcs_version = "PY2AND3",
     tags = [