fix

Signed-off-by: Chad Dombrova <chadrik@gmail.com>

Author: chadrik

.github/workflows/wheel.yml

@@ -147,6 +147,8 @@ jobs:
     steps:
       - name: Checkout repo
         uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
+        with:
+          ref: ${{ github.event.pull_request.head.ref }}
 
       - name: Install Python
         uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # v5.2.0
@@ -171,11 +173,24 @@ jobs:
             ./wheelhouse/*.whl
             ./wheelhouse/OpenImageIO/__init__.pyi
         # if stub validation fails we want to upload the stubs for users to review
-        if: success() || failure()
-
-      # Commit all changed files back to the repository
-      - uses: stefanzweifel/git-auto-commit-action@v6
-        commit_message: "Automatic update to python stubs"
+#         if: success() || failure()
+
+      - name: Copy stubs to repo
+        run: |
+          pwd
+          ls -la .
+          ls -la ./wheelhouse/
+          if [ -f ./wheelhouse/OpenImageIO/__init__.pyi ]; then
+            echo "Copying stubs into repo"
+            cp ./wheelhouse/OpenImageIO/__init__.pyi ./src/python/stubs/OpenImageIO/__init__.pyi
+          fi
+#         if: failure()
+
+      - name: Commit changed stubs back to the repository
+        uses: stefanzweifel/git-auto-commit-action@v6
+        with:
+          commit_message: "Automatic update to python stubs"
+#         if: failure()
 
   # ---------------------------------------------------------------------------
   # Linux ARM Wheels

src/python/stubs/generate_stubs.py

@@ -209,7 +209,7 @@ def main() -> None:
             print(get_colored_diff(old_text, new_text))
             print("Run `make pystubs` locally and commit the results for review.")
             print("The resulting __init__.pyi file will be uploaded as an artifact on this job.")
-            sys.exit(2)
+            # sys.exit(2)
 
 
 if __name__ == "__main__":