From 4b33b00882af9b702409303a1ddbdfe99dea063b Mon Sep 17 00:00:00 2001
From: "Thomas J. Fan" <thomasjpfan@gmail.com>
Date: Tue, 12 Nov 2024 16:01:18 +0000
Subject: [PATCH] Revert "Block on Python 3.13 version (#1899)"

This reverts commit 8e013c22fd9645b97d1c2c0bf5855455bf185279.
---
 .github/workflows/build_linux_wheels.yaml | 34 ++---------------------
 1 file changed, 3 insertions(+), 31 deletions(-)

diff --git a/.github/workflows/build_linux_wheels.yaml b/.github/workflows/build_linux_wheels.yaml
index ab0042c682..5a2605db11 100644
--- a/.github/workflows/build_linux_wheels.yaml
+++ b/.github/workflows/build_linux_wheels.yaml
@@ -27,46 +27,18 @@ jobs:
       with-cuda: enable
       with-rocm: enable
       build-python-only: enable
-  # TODO: Remove `filter-python-version` after PyArrow releases v18
-  filter-python-versions:
-    needs: generate-matrix
-    runs-on: ubuntu-latest
-    outputs:
-      matrix: ${{ steps.set-matrix.outputs.matrix }}
-    steps:
-      - name: Filter matrix to exclude Python 3.13
-        id: set-matrix
-        shell: python
-        env:
-          input-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-        run: |
-          import os
-          import json
-
-          # Grab environment variables
-          input_matrix = json.loads(os.environ["input-matrix"])
-          github_output_file = os.environ["GITHUB_OUTPUT"]
-
-          # Filter out any builds for 3.13
-          filtered_matrix = {"include": []}
-          for build in input_matrix["include"]:
-            if build["python_version"] != "3.13":
-              filtered_matrix["include"].append(build)
-
-          # Write the new matrix to the default outputs file
-          with open(github_output_file, "w") as handle:
-            handle.write(f"matrix={json.dumps(filtered_matrix)}")
   build:
-    needs: filter-python-versions
+    needs: generate-matrix
     name: ${{ matrix.repository }}
     uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
+    if: ${{ needs.generate-matrix.outputs.matrix.python_version }} != '3.13'
     strategy:
       fail-fast: false
     with:
       repository: pytorch/torchtune
       ref: ""
       package-name: torchtune
-      build-matrix: ${{ needs.filter-python-versions.outputs.matrix }}
+      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: .github/scripts/pre_build_script.sh
       trigger-event: ${{ github.event_name }}
       build-platform: 'python-build-package'