From cf3bb7ff1d602ab7f40dbc8156c873c3ddcac8b9 Mon Sep 17 00:00:00 2001
From: Harshal Sheth <hsheth2@gmail.com>
Date: Wed, 1 Feb 2023 18:30:18 -0500
Subject: [PATCH] ci(): fix datahub-upgrade quickstart regression (#7217)

Co-authored-by: John Joyce <john@acryl.io>
---
 .github/workflows/docker-unified.yml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/.github/workflows/docker-unified.yml b/.github/workflows/docker-unified.yml
index 6d0e4213bfae87..54160b71d78645 100644
--- a/.github/workflows/docker-unified.yml
+++ b/.github/workflows/docker-unified.yml
@@ -434,6 +434,7 @@ jobs:
         elasticsearch_setup_build,
         mae_consumer_build,
         mce_consumer_build,
+        datahub_upgrade_build,
       ]
     steps:
       - name: Check out the repo
@@ -487,6 +488,11 @@ jobs:
         if: ${{ needs.setup.outputs.publish != 'true' }}
         with:
           image: ${{ env.DATAHUB_MAE_CONSUMER_IMAGE }}:${{ needs.setup.outputs.unique_tag }}
+      - name: Download upgrade image
+        uses: ishworkh/docker-image-artifact-download@v1
+        if: ${{ needs.setup.outputs.publish != 'true' }}
+        with:
+          image: ${{ env.DATAHUB_UPGRADE_IMAGE }}:${{ needs.setup.outputs.unique_tag }}
       - name: Calculate pip install plan
         id: "packages_checksum"
         run: |