From 600b43e12dcc910fe85f1925385e2d49fb530437 Mon Sep 17 00:00:00 2001
From: "Peter St. John" <pstjohn@nvidia.com>
Date: Thu, 2 Jan 2025 10:25:04 -0800
Subject: [PATCH] don't cleanup until end

Signed-off-by: Peter St. John <pstjohn@nvidia.com>
---
 .github/workflows/unit-tests.yml | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
index b9f5455914..77b2733580 100644
--- a/.github/workflows/unit-tests.yml
+++ b/.github/workflows/unit-tests.yml
@@ -40,9 +40,6 @@ jobs:
   build-bionemo-image:
     needs: pre-commit
     runs-on: self-hosted-nemo-gpus-1  # TODO: make this a CPU-only builder runner.
-    defaults:
-      run:
-        working-directory: ./${{ github.run_id }}
     steps:
       - name: Checkout repository
         uses: actions/checkout@v4
@@ -112,11 +109,13 @@ jobs:
 
   teardown-environment:
     runs-on: ubuntu-latest
+    needs: run-tests
+    if: always()
     steps:
       - name: Docker Prune
         run: |
             docker system prune $([[ "$LABEL" != "" ]] && echo --filter "label=nemo.library=$LABEL" || echo '') --filter "until=168h" --force
-      - name: Create working directory
+      - name: Delete working directory
         run: |
             rm -rf ./${{ github.run_id }}