Add test output cleanup

Signed-off-by: Igor Gitman <[email protected]>

.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