From 448a4e0c31c05edf72e3bf84203c0e3e4a889dc9 Mon Sep 17 00:00:00 2001 From: Igor Gitman Date: Thu, 2 Jan 2025 10:50:28 -0800 Subject: [PATCH] Add test output cleanup Signed-off-by: Igor Gitman --- .github/workflows/gpu_tests.yml | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/.github/workflows/gpu_tests.yml b/.github/workflows/gpu_tests.yml index c091a387a..9ccd89624 100644 --- a/.github/workflows/gpu_tests.yml +++ b/.github/workflows/gpu_tests.yml @@ -46,6 +46,10 @@ jobs: nvidia-smi set -o pipefail # this will make sure next line returns non-0 exit code if tests fail ./tests/gpu-tests/run_llama.sh + - name: Cleanup test directory + if: always() + run: | + rm -rf /tmp/nemo-skills-tests/llama gpu-tests-qwen: runs-on: self-hosted-nemo-gpus-1 @@ -77,7 +81,11 @@ jobs: nvidia-smi set -o pipefail # this will make sure next line returns non-0 exit code if tests fail ./tests/gpu-tests/run_qwen.sh - + - name: Cleanup test directory + if: always() + run: | + rm -rf /tmp/nemo-skills-tests/qwen + gpu-tests-rm: runs-on: self-hosted-nemo-gpus-1 if: ${{ github.event.label.name == 'run GPU tests' }} @@ -107,3 +115,7 @@ jobs: nvidia-smi set -o pipefail # this will make sure next line returns non-0 exit code if tests fail ./tests/gpu-tests/run_rm.sh + - name: Cleanup test directory + if: always() + run: | + rm -rf /tmp/nemo-skills-tests/mistral_emb