diff --git a/.github/workflows/publish_python_sdk.yml b/.github/workflows/publish_python_sdk.yml
index 5866eba479..ab6f97873d 100644
--- a/.github/workflows/publish_python_sdk.yml
+++ b/.github/workflows/publish_python_sdk.yml
@@ -25,23 +25,20 @@ on:
         type: string
 
 jobs:
+  build-wheels:
+    uses: ./.github/workflows/build_wheels.yml
+    with:
+      custom_version: ${{ github.event.inputs.custom_version }}
+      token: ${{ github.event.inputs.token }}
+
   publish-python-sdk:
     if: github.repository == 'feast-dev/feast'
     runs-on: ubuntu-latest
+    needs: [ build-wheels ]
     steps:
       - uses: actions/checkout@v4
         with:
           submodules: 'true'
-      - id: get-version
-        uses: ./.github/actions/get-semantic-release-version
-        with:
-          custom_version: ${{ github.event.inputs.custom_version }}
-          token: ${{ github.event.inputs.token }}
-      - uses: ./.github/workflows/build_wheels.yml
-        with:
-          release_version: ${{ steps.get-version.outputs.release_version }}
-          highest_semver_tag: ${{ steps.get-version.outputs.highest_semver_tag }}
-          version_without_prefix: ${{ steps.get-version.outputs.version_without_prefix }}
       - uses: actions/download-artifact@v4.1.7
         with:
           name: python-wheels