Exclude enormous-sized tests from the CI

//tensorflow_model_optimization/python/core/quantization/keras:quantize_models_test uses more RAM that the CI machine provides, causing long build times and timeout failures.

PiperOrigin-RevId: 415435354

Author: fredrec

ci/kokoro/build.sh

@@ -28,6 +28,9 @@ set -o pipefail  # Treat the failure of a command in a pipeline as error.
 #  parameters, will print the full command, with credentials, in the build logs.
 # set -x
 
-pip3 install --requirement "requirements.txt"
+pip install --requirement "requirements.txt"
 
-bazel test --python_path=$PYTHON_BIN //tensorflow_model_optimization/python/core/...
+# Run the tests.
+# Some tests requiring more RAM that the CI machine provides are disabled.
+bazel test --test_size_filters="-enormous" \
+  //tensorflow_model_optimization/python/core/...

tensorflow_model_optimization/python/core/quantization/keras/BUILD

@@ -297,7 +297,7 @@ py_strict_test(
 
 py_strict_test(
     name = "quantize_models_test",
-    size = "large",
+    size = "enormous",
     srcs = ["quantize_models_test.py"],
     flaky = True,
     python_version = "PY3",