From a06cfdc7e4086e51d75fea94aea191a59118218b Mon Sep 17 00:00:00 2001
From: pavelzbornik <12965149+pavelzbornik@users.noreply.github.com>
Date: Fri, 22 Nov 2024 12:28:58 +0000
Subject: [PATCH] refactor: rename build step and streamline Docker image
 handling in CI workflow

---
 .github/workflows/CI.yaml | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/CI.yaml b/.github/workflows/CI.yaml
index 9c84ad9..8376c22 100644
--- a/.github/workflows/CI.yaml
+++ b/.github/workflows/CI.yaml
@@ -31,17 +31,15 @@ jobs:
         docker-images: false
         swap-storage: true
 
-    - name: Build Docker image
+    - name: Build and export to Docker
       uses: docker/build-push-action@v6
       with:
         context: .
+        load: true
         push: false
         tags: whisperx-fastapi:latest
         cache-from: type=gha
         cache-to: type=gha,mode=max
 
-    - name: Load Docker image
-      run: docker load -i $(docker save whisperx-fastapi:latest -o whisperx-fastapi.tar)
-
     - name: Run tests
-      run: docker run --env MY_ENV_VAR=${{ secrets.HF_TOKEN }} --env WHISPER_MODEL=tiny --env DEFAULT_LANG=en --entrypoint pytest whisperx-fastapi:latest
+      run: docker run --rm --env MY_ENV_VAR=${{ secrets.HF_TOKEN }} --env WHISPER_MODEL=tiny --env DEFAULT_LANG=en --entrypoint pytest whisperx-fastapi:latest