Clean pytest-dist

tools/testing/build_and_run_tests.sh

@@ -25,14 +25,5 @@ python ./configure.py
 bash tools/install_so_files.sh
 python -c "import tensorflow as tf; print(tf.config.list_physical_devices())"
 
-# use 10 workers if a gpu is available, otherwise,
-# one worker per cpu core. Kokoro has 38 cores, that'd be too much
-# for the gpu memory, until we change the device placement to
-# use multiple gpus when they are available.
-EXTRA_ARGS="-n 10"
-if ! [ -x "$(command -v nvidia-smi)" ]; then
-  EXTRA_ARGS="-n auto"
-fi
-
 bazel clean
-python -m pytest -v --functions-durations=20 --modules-durations=5 $SKIP_CUSTOM_OP_TESTS_FLAG $EXTRA_ARGS ./tensorflow_addons
+python -m pytest -v --functions-durations=20 --modules-durations=5 $SKIP_CUSTOM_OP_TESTS_FLAG ./tensorflow_addons