Unverified Commit 87726a08 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

`pytest_num_workers=4` for some CircleCI jobs (#31764)



pytest_num_workers=4
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 048f599f
...@@ -248,7 +248,7 @@ torch_job = CircleCIJob( ...@@ -248,7 +248,7 @@ torch_job = CircleCIJob(
docker_image=[{"image": "huggingface/transformers-torch-light"}], docker_image=[{"image": "huggingface/transformers-torch-light"}],
install_steps=["uv venv && uv pip install ."], install_steps=["uv venv && uv pip install ."],
parallelism=6, parallelism=6,
pytest_num_workers=16 pytest_num_workers=4
) )
tokenization_job = CircleCIJob( tokenization_job = CircleCIJob(
...@@ -256,7 +256,7 @@ tokenization_job = CircleCIJob( ...@@ -256,7 +256,7 @@ tokenization_job = CircleCIJob(
docker_image=[{"image": "huggingface/transformers-torch-light"}], docker_image=[{"image": "huggingface/transformers-torch-light"}],
install_steps=["uv venv && uv pip install ."], install_steps=["uv venv && uv pip install ."],
parallelism=6, parallelism=6,
pytest_num_workers=16 pytest_num_workers=4
) )
...@@ -265,7 +265,7 @@ tf_job = CircleCIJob( ...@@ -265,7 +265,7 @@ tf_job = CircleCIJob(
docker_image=[{"image":"huggingface/transformers-tf-light"}], docker_image=[{"image":"huggingface/transformers-tf-light"}],
install_steps=["uv venv", "uv pip install -e."], install_steps=["uv venv", "uv pip install -e."],
parallelism=6, parallelism=6,
pytest_num_workers=16, pytest_num_workers=4,
) )
...@@ -274,7 +274,7 @@ flax_job = CircleCIJob( ...@@ -274,7 +274,7 @@ flax_job = CircleCIJob(
docker_image=[{"image":"huggingface/transformers-jax-light"}], docker_image=[{"image":"huggingface/transformers-jax-light"}],
install_steps=["uv venv && uv pip install ."], install_steps=["uv venv && uv pip install ."],
parallelism=6, parallelism=6,
pytest_num_workers=16 pytest_num_workers=4
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment