Update update-ttnn-wheel.yaml

tenstorrent · Feb 9, 2025 · f4b0287 · f4b0287
1 parent fc139d8
commit f4b0287
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/.github/workflows/update-ttnn-wheel.yaml b/.github/workflows/update-ttnn-wheel.yaml
@@ -28,10 +28,13 @@ jobs:
         id: update-requirements
         run: |
           latest_version=${{ steps.fetch_release.outputs.release }}
-          latest_version_short=$(echo $latest_version | sed 's/-rc/rc/')          
-          sed -i '/^https:\/\/github\.com\/tenstorrent\/tt-metal\/releases\//d' requirements.txt
-          echo "ttnn@https://github.com/tenstorrent/tt-metal/releases/download/v$latest_version/ttnn-$latest_version_short+any-cp38-cp38-linux_x86_64.whl" >> requirements.txt
-
+          # Remove any existing ttnn lines (adjust the regex if needed)
+          sed -i '/^ttnn @ https:\/\/github\.com\/tenstorrent\/tt-metal\/releases\//d' requirements.txt
+      
+          # Append the two lines for the different python versions.
+          echo "ttnn @ https://github.com/tenstorrent/tt-metal/releases/download/v${latest_version}/ttnn-$latest_version_short+any-cp38-cp38-linux_x86_64.whl ; python_version==\"3.8\"" >> requirements.txt
+          echo "ttnn @ https://github.com/tenstorrent/tt-metal/releases/download/v${latest_version}/ttnn-$latest_version_short+any-cp310-cp310-linux_x86_64.whl ; python_version==\"3.10\"" >> requirements.txt
+
       - name: Create Pull Request
         uses: peter-evans/create-pull-request@v7
         id: create-pr