message: 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).
diff: @@ -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",
  