From 87726a08ed12e74ddc078ea877d9decf9a30cb8b Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 4 Jul 2024 14:44:58 +0200 Subject: [PATCH] `pytest_num_workers=4` for some CircleCI jobs (#31764) pytest_num_workers=4 Co-authored-by: ydshieh --- .circleci/create_circleci_config.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index d6d1041ea6..d783488cae 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -248,7 +248,7 @@ torch_job = CircleCIJob( docker_image=[{"image": "huggingface/transformers-torch-light"}], install_steps=["uv venv && uv pip install ."], parallelism=6, - pytest_num_workers=16 + pytest_num_workers=4 ) tokenization_job = CircleCIJob( @@ -256,7 +256,7 @@ tokenization_job = CircleCIJob( docker_image=[{"image": "huggingface/transformers-torch-light"}], install_steps=["uv venv && uv pip install ."], parallelism=6, - pytest_num_workers=16 + pytest_num_workers=4 ) @@ -265,7 +265,7 @@ tf_job = CircleCIJob( docker_image=[{"image":"huggingface/transformers-tf-light"}], install_steps=["uv venv", "uv pip install -e."], parallelism=6, - pytest_num_workers=16, + pytest_num_workers=4, ) @@ -274,7 +274,7 @@ flax_job = CircleCIJob( docker_image=[{"image":"huggingface/transformers-jax-light"}], install_steps=["uv venv && uv pip install ."], parallelism=6, - pytest_num_workers=16 + pytest_num_workers=4 )