From 7e48852d83f666089bfe351882ffe3570fc2bb8a Mon Sep 17 00:00:00 2001 From: wenxindongwork Date: Mon, 10 Nov 2025 17:06:59 +0000 Subject: [PATCH] wip Signed-off-by: wenxindongwork --- .buildkite/pipeline_jax.yml | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/.buildkite/pipeline_jax.yml b/.buildkite/pipeline_jax.yml index 79787592f..cf4e5519b 100644 --- a/.buildkite/pipeline_jax.yml +++ b/.buildkite/pipeline_jax.yml @@ -229,8 +229,13 @@ steps: queue: tpu_v6e_8_queue commands: - | - .buildkite/scripts/run_in_docker.sh \ - bash -c 'python3 -m pytest -s -v -x /workspace/tpu_inference/tests/e2e/test_data_parallel.py' + if [[ "$$NIGHTLY" == "1" ]]; then + .buildkite/scripts/run_in_docker.sh \ + bash -c 'python3 -m pytest -s -v -x /workspace/tpu_inference/tests/e2e/test_data_parallel.py' + else + echo "Skipping: NIGHTLY environment variable not set" + exit 0 + fi - label: "lora unit tests on single chip" key: test_15