fix pypi build

Author: zhangfengcdt

.github/workflows/pypi.yml

@@ -8,23 +8,56 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        os: [ubuntu-latest, windows-latest, macos-latest, macos-14]  # macos-14 is ARM64
+        include:
+          # Linux
+          - os: ubuntu-latest
+            target: x86_64
+            manylinux: auto
+#          - os: ubuntu-latest
+#            target: aarch64
+#            manylinux: auto
+#          # Windows
+#          - os: windows-latest
+#            target: x64
+#            manylinux: false
+#          - os: windows-latest
+#            target: x86
+#            manylinux: false
+          # macOS
+          - os: macos-latest
+            target: x86_64
+            manylinux: false
+#          - os: macos-14
+#            target: aarch64
+#            manylinux: false
 
     steps:
     - uses: actions/checkout@v4
 
+    - name: Setup Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: |
+          3.11
+
     - name: Build wheels
       uses: PyO3/maturin-action@v1
       with:
         target: ${{ matrix.target }}
         args: --release --out dist --features python --find-interpreter
         sccache: 'true'
-        manylinux: auto
+        manylinux: ${{ matrix.manylinux }}
+        rust-toolchain: stable
+        before-script-linux: |
+          # Install any system dependencies if needed
+          if [[ "${{ matrix.target }}" == "aarch64" ]]; then
+            echo "Setting up cross-compilation for aarch64"
+          fi
 
     - name: Upload wheels
       uses: actions/upload-artifact@v4
       with:
-        name: wheels-${{ matrix.os }}
+        name: wheels-${{ matrix.os }}-${{ matrix.target }}
         path: dist
 
   build-sdist:
@@ -33,22 +66,53 @@ jobs:
     steps:
     - uses: actions/checkout@v4
 
+    - name: Setup Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.11'
+
     - name: Build sdist
       uses: PyO3/maturin-action@v1
       with:
         command: sdist
         args: --out dist
+        rust-toolchain: stable
 
     - name: Upload sdist
       uses: actions/upload-artifact@v4
       with:
         name: wheels-sdist
         path: dist
 
+  test-wheel:
+    name: Test built wheel
+    runs-on: ubuntu-latest
+    needs: [build-wheels]
+    
+    steps:
+    - uses: actions/checkout@v4
+    
+    - name: Setup Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.11'
+    
+    - name: Download Linux wheel
+      uses: actions/download-artifact@v4
+      with:
+        name: wheels-ubuntu-latest-x86_64
+        path: dist
+    
+    - name: Install and test wheel
+      run: |
+        pip install dist/*.whl
+        python -c "from prollytree import ProllyTree; tree = ProllyTree(); tree.insert(b'test', b'value'); assert tree.find(b'test') == b'value'"
+        echo "✅ Wheel test passed!"
+
   collect-artifacts:
     name: Collect all artifacts
     runs-on: ubuntu-latest
-    needs: [build-wheels, build-sdist]
+    needs: [build-wheels, build-sdist, test-wheel]
 
     steps:
     - name: Download all artifacts
@@ -64,6 +128,9 @@ jobs:
         ls -la dist/
         echo ""
         echo "Total artifacts: $(ls -1 dist/ | wc -l)"
+        echo ""
+        echo "Artifact details:"
+        du -h dist/*
 
     - name: Upload combined artifacts
       uses: actions/upload-artifact@v4