add skip wheel install fact (#185)

We want to only skip the tt-forge wheel install test for now. I have
added a release fact `skip_wheel_install` boolean to control this
behavior for easy future adjustment.

Fixes
https://github.com/tenstorrent/tt-forge/actions/runs/15860335527/job/44715803236#step:3:329


```bash
The conflict is caused by:
    pjrt-plugin-tt 0.1.0.dev20250624 depends on jax>=0.6.0
    tt-forge-fe 0.1.0.dev20250624 depends on jax==0.4.30
```

Author: jmcgrathTT

.github/actions/publish-tenstorrent-pypi/action.yml

@@ -81,12 +81,14 @@ runs:
         for wheel_name in $pip_wheel_names; do
           # Check the only tenstorrent index for the wheel version
           pip index versions $wheel_name --pre --index-url https://pypi.eng.aws.tenstorrent.com/ | grep ${{ inputs.new_version_tag }}
-          # Test install wheel version.
-          pip install $wheel_name==${{ inputs.new_version_tag }} --no-cache-dir --extra-index-url https://pypi.eng.aws.tenstorrent.com/
-          # Wheel import wheel
-          # TODO: skip import test until wheels are fixed
-          #normalized_import=$(echo "$wheel_name" | sed -e 's/-/_/g')
-          #python -c "import $normalized_import"
-          # Check installed version of wheel
-          pip show $wheel_name | grep ${{ inputs.new_version_tag }}
+          if [ "${{ steps.set-release-facts.outputs.skip_wheel_install }}" == "false" ]; then
+            # Test install wheel version.
+            pip install $wheel_name==${{ inputs.new_version_tag }} --no-cache-dir --extra-index-url https://pypi.eng.aws.tenstorrent.com/
+            # Wheel import wheel
+            # TODO: skip import test until wheels are fixed
+            #normalized_import=$(echo "$wheel_name" | sed -e 's/-/_/g')
+            #python -c "import $normalized_import"
+            # Check installed version of wheel
+            pip show $wheel_name | grep ${{ inputs.new_version_tag }}
+          fi
         done

.github/actions/set-release-facts/action.yaml

@@ -42,6 +42,10 @@ outputs:
   pip_wheel_deps_names:
     description: "Names of pip wheel dependencies"
     value: ${{ steps.set-manifest.outputs.pip_wheel_deps_names }}
+  skip_wheel_install:
+    description: "Skip wheel install"
+    value: ${{ steps.set-manifest.outputs.skip_wheel_install }}
+
 runs:
   using: "composite"
   steps:
@@ -50,6 +54,10 @@ runs:
       shell: bash
       run: |
         # Set release facts based on repo
+
+        # Globals
+        skip_wheel_install="false"
+
         if [[ "${{ inputs.repo }}" =~ "tt-forge-fe" ]]; then
             workflow="On nightly"
             if [[ "${{ inputs.release_type }}" == "nightly" ]]; then
@@ -86,6 +94,7 @@ runs:
         elif [[ "${{ inputs.repo }}" =~ "tt-forge" ]]; then
             pip_wheel_deps_names="tt-torch tt_forge_fe tt_tvm pjrt-plugin-tt"
             pip_wheel_names="tt-forge"
+            skip_wheel_install="true"
         else
             echo "Unknown repo: ${{ inputs.repo }}"
             exit 1
@@ -112,6 +121,7 @@ runs:
         echo "Using repo_short: $repo_short"
         echo "Using pip_wheel_names: $pip_wheel_names"
         echo "Using pip_wheel_deps_names: $pip_wheel_deps_names"
+        echo "Using skip_wheel_install: $skip_wheel_install"
 
         # Set outputs
         echo "workflow=$workflow" >> $GITHUB_OUTPUT
@@ -123,3 +133,4 @@ runs:
         echo "new_version_tag=$new_version_tag" >> $GITHUB_OUTPUT
         echo "pip_wheel_names=$pip_wheel_names" >> $GITHUB_OUTPUT
         echo "pip_wheel_deps_names=$pip_wheel_deps_names" >> $GITHUB_OUTPUT
+        echo "skip_wheel_install=$skip_wheel_install" >> $GITHUB_OUTPUT

.github/actions/tt-forge-wheel/action.yml

@@ -58,4 +58,8 @@ runs:
         mv ${{ github.workspace }}/dist/*.whl ${{ github.workspace }}/release/artifacts/.
         cd ${{ github.workspace }}/release/artifacts
         # TODO: build alerting around this.
-        pip install *.whl || echo "Failed to install wheel"
+        if [ "${{ steps.set-release-facts.outputs.skip_wheel_install }}" == "false" ]; then
+          pip install *.whl
+        else
+          pip install *.whl || echo "Failed to install wheel"
+        fi