@@ -27,46 +27,18 @@ jobs:
27
27
with-cuda : enable
28
28
with-rocm : enable
29
29
build-python-only : enable
30
- # TODO: Remove `filter-python-version` after PyArrow releases v18
31
- filter-python-versions :
32
- needs : generate-matrix
33
- runs-on : ubuntu-latest
34
- outputs :
35
- matrix : ${{ steps.set-matrix.outputs.matrix }}
36
- steps :
37
- - name : Filter matrix to exclude Python 3.13
38
- id : set-matrix
39
- shell : python
40
- env :
41
- input-matrix : ${{ needs.generate-matrix.outputs.matrix }}
42
- run : |
43
- import os
44
- import json
45
-
46
- # Grab environment variables
47
- input_matrix = json.loads(os.environ["input-matrix"])
48
- github_output_file = os.environ["GITHUB_OUTPUT"]
49
-
50
- # Filter out any builds for 3.13
51
- filtered_matrix = {"include": []}
52
- for build in input_matrix["include"]:
53
- if build["python_version"] != "3.13":
54
- filtered_matrix["include"].append(build)
55
-
56
- # Write the new matrix to the default outputs file
57
- with open(github_output_file, "w") as handle:
58
- handle.write(f"matrix={json.dumps(filtered_matrix)}")
59
30
build :
60
- needs : filter-python-versions
31
+ needs : generate-matrix
61
32
name : ${{ matrix.repository }}
62
33
uses : pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
34
+ if : ${{ needs.generate-matrix.outputs.matrix.python_version }} != '3.13'
63
35
strategy :
64
36
fail-fast : false
65
37
with :
66
38
repository : pytorch/torchtune
67
39
ref : " "
68
40
package-name : torchtune
69
- build-matrix : ${{ needs.filter-python-versions .outputs.matrix }}
41
+ build-matrix : ${{ needs.generate-matrix .outputs.matrix }}
70
42
pre-script : .github/scripts/pre_build_script.sh
71
43
trigger-event : ${{ github.event_name }}
72
44
build-platform : ' python-build-package'
0 commit comments