message: Remove small size annotation from discretization test
  These timeout rarely and should run as medium. Also remove medium annotations from
  other tests (this is default).
diff: @@ -334,7 +334,6 @@ py_library(
  cuda_py_test(
  name = "category_crossing_test",
  - size = "medium",
  srcs = ["category_crossing_test.py"],
  python_version = "PY3",
  shard_count = 4,
  @@ -362,7 +361,6 @@ cuda_py_test(
  tf_py_test(
  name = "category_encoding_test",
  - size = "medium",
  srcs = ["category_encoding_test.py"],
  python_version = "PY3",
  deps = [
  @@ -456,7 +454,6 @@ distribute_py_test(
  tf_py_test(
  name = "discretization_test",
  - size = "small",
  srcs = ["discretization_test.py"],
  python_version = "PY3",
  shard_count = 4,
  @@ -492,7 +489,6 @@ distribute_py_test(
  cuda_py_test(
  name = "hashing_test",
  - size = "medium",
  srcs = ["hashing_test.py"],
  python_version = "PY3",
  shard_count = 4,
  @@ -531,7 +527,6 @@ distribute_py_test(
  tf_py_test(
  name = "index_lookup_test",
  - size = "medium",
  srcs = ["index_lookup_test.py"],
  python_version = "PY3",
  tags = ["noasan"], # TODO(b/183961255)
  @@ -566,7 +561,6 @@ distribute_py_test(
  cuda_py_test(
  name = "image_preprocessing_test",
  - size = "medium",
  srcs = ["image_preprocessing_test.py"],
  python_version = "PY3",
  shard_count = 4,
  @@ -610,7 +604,6 @@ tf_py_test(
  tf_py_test(
  name = "integer_lookup_test",
  - size = "medium",
  srcs = ["integer_lookup_test.py"],
  python_version = "PY3",
  tags = ["noasan"], # TODO(b/183961255)
  @@ -662,7 +655,6 @@ tf_py_test(
  tf_py_test(
  name = "text_vectorization_test",
  - size = "medium",
  srcs = ["text_vectorization_test.py"],
  python_version = "PY3",
  shard_count = 4,
  @@ -718,7 +710,6 @@ tf_py_test(
  tf_py_test(
  name = "string_lookup_test",
  - size = "medium",
  srcs = ["string_lookup_test.py"],
  python_version = "PY3",
  tags = [
  