diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index 0fcde009ed8..5696915dc80 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD @@ -873,10 +873,9 @@ cuda_py_test( srcs = ["remote_execution_test.py"], grpc_enabled = True, python_version = "PY3", - shard_count = 2, + shard_count = 8, tags = [ "no_oss", # This test launches local server - "notsan", # TODO(b/151240519) Re-enable. ], deps = [ "//tensorflow/python:array_ops",