From 8037d69188664bf02f3bdca5b6cb376da71fc696 Mon Sep 17 00:00:00 2001
From: Eduardo Apolinario <eapolinario@users.noreply.github.com>
Date: Tue, 17 Dec 2024 15:39:35 -0500
Subject: [PATCH] Use correct name in flyteagent test

Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>
---
 .github/workflows/pythonpublish.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/pythonpublish.yml b/.github/workflows/pythonpublish.yml
index e5a8118923..4aceee472e 100644
--- a/.github/workflows/pythonpublish.yml
+++ b/.github/workflows/pythonpublish.yml
@@ -224,7 +224,7 @@ jobs:
           cache-to: type=gha,mode=max
       - name: Confirm Agent can start
         run: |
-          docker run --rm ghcr.io/${{ github.repository_owner }}/flyteagent:${{ github.sha }} pyflyte serve agent --port 8000 --timeout 1
+          docker run --rm ghcr.io/${{ github.repository_owner }}/flyteagent-slim:${{ github.sha }} pyflyte serve agent --port 8000 --timeout 1
       - name: Push flyteagent-all Image to GitHub Registry
         uses: docker/build-push-action@v2
         with: