chore: Updating build_wheels to accept input from publish workflow

Signed-off-by: Francisco Javier Arceo <[email protected]>

.github/workflows/build_wheels.yml

@@ -8,6 +8,11 @@ on:
       tags:
         - 'v*.*.*'
   workflow_call:
+    inputs:
+      release_version:
+        description: 'The release version to use (e.g., v1.2.3)'
+        required: true
+        type: string
 
 jobs:
   get-version:
@@ -23,7 +28,14 @@ jobs:
           persist-credentials: false
       - name: Get release version
         id: get_release_version
-        run: echo ::set-output name=release_version::${GITHUB_REF#refs/*/}
+        run: |
+          if [[ -n "${{ inputs.release_version }}" ]]; then
+            echo "Using provided release version: ${{ inputs.release_version }}"
+            echo "::set-output name=release_version::${{ inputs.release_version }}"
+          else
+            echo "No release version provided. Falling back to GITHUB_REF."
+            echo "::set-output name=release_version::${GITHUB_REF#refs/tags/}"
+          fi
       - name: Get release version without prefix
         id: get_release_version_without_prefix
         env:

.github/workflows/publish.yml

@@ -42,6 +42,7 @@ jobs:
             fi
             echo "Validated custom version: ${{ github.event.inputs.custom_version }}"
           fi
+
       - name: Get release version
         id: get_release_version
         run: |
@@ -169,6 +170,8 @@ jobs:
 
   build_wheels:
     uses: ./.github/workflows/build_wheels.yml
+    with:
+      release_version: ${{ github.event.inputs.custom_version }}
 
   publish-python-sdk:
     if: github.repository == 'feast-dev/feast'