message: Deflake optimize_dataset_test in tsan mode.
diff: @@ -286,7 +286,7 @@ py_test(
  py_test(
  name = "optimize_dataset_test",
  - size = "small",
  + size = "medium",
  srcs = ["optimize_dataset_test.py"],
  srcs_version = "PY2AND3",
  tags = [
  