add multiple python version

Signed-off-by: Austin Liu <[email protected]>
flyteorg · Aug 19, 2024 · 57cdf20 · 57cdf20
1 parent 92720c2
commit 57cdf20
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 10 deletions.
diff --git a/.github/workflows/flyteidl-nightly.yml b/.github/workflows/flyteidl-nightly.yml
@@ -27,32 +27,37 @@ jobs:
         platform:
           - runner: ubuntu-latest
             target: x86_64
-          # - runner: ubuntu-latest
-          #   target: x86
-          # - runner: ubuntu-latest
-          #   target: aarch64
           - runner: ubuntu-latest
             target: armv7
-          # - runner: ubuntu-latest
-          #   target: s390x
           - runner: ubuntu-latest
             target: ppc64le
+        python-version: 
+          - 3.9
+          - 3.10
+          - 3.11
+          - 3.12
     steps:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: 3.x
+          python-version: ${{ matrix.python-version }}
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:
           target: ${{ matrix.platform.target }}
           command: build
           args: --release --out dist --find-interpreter -m flyteidl/Cargo.toml
-          before-script-linux: sudo apt update && sudo apt install -y protobuf-compiler libprotobuf-dev && cd ${{ github.workspace }}/flyteidl && rm -rf ./gen/pb_rust/* && cargo run --bin gen_flyteidl && cd ../
+          before-script-linux: |
+            sudo apt update && sudo apt install -y protobuf-compiler libprotobuf-dev
+            cd ${{ github.workspace }}/flyteidl
+            rm -rf ./gen/pb_rust/*
+            cargo run --bin gen_flyteidl
+            cd ../
+            export PYO3_CROSS_PYTHON_VERSION=${{ matrix.python-version }}
       - name: Upload wheels
         uses: actions/upload-artifact@v4
         with:
-          name: wheels-linux-${{ matrix.platform.target }}
+          name: wheels-linux-${{ matrix.platform.target }}-py${{ matrix.python-version }}
           path: dist
 
   musllinux:

diff --git a/flyteidl/Cargo.toml b/flyteidl/Cargo.toml
@@ -36,7 +36,7 @@ tonic-build = "0.11.0"
 tonic ={ version = "0.11.0", features = ["tls", "tls-roots"] }
 glob = "0.3.1"
 pyo3_macro = { path = "./pyo3_macro" }
-pyo3 =  { version = "0.22.1", features = [ "extension-module", "multiple-pymethods", "experimental-async", "gil-refs", "experimental-inspect", "abi3-py39" ] }
+pyo3 =  { version = "0.22.1", features = [ "extension-module", "multiple-pymethods", "experimental-async", "gil-refs", "experimental-inspect" ] }
 tokio = { version = "1.0", features = ["macros", "rt-multi-thread"] }
 serde_json = "1.0.120"
 serde = { version = "1.0.204", features = ["serde_derive"]}