misc: release v0.3.1 (#42)

* misc: update dependencies

* misc: release v0.3.1

* misc: format code

* fix: pyo3 extension-module

* fix: ci python macos working-directory

Author: ChieloNewctle

.github/workflows/ci-python.yml

@@ -1,17 +1,13 @@
 name: Build & Test the Python Bindings
-
 defaults:
   run:
     working-directory: python
-
 on:
   push:
   pull_request:
   workflow_dispatch:
-
 permissions:
   contents: read
-
 jobs:
   format:
     name: Check Python format
@@ -28,7 +24,6 @@ jobs:
         run: ruff check .
       - name: Black
         run: black --check --diff .
-
   rustfmt:
     name: Check Rust format
     runs-on: ubuntu-latest
@@ -39,14 +34,12 @@ jobs:
       - run: rustup update stable && rustup default stable
       - run: rustup component add rustfmt
       - run: cargo fmt --all --check
-
   test:
     name: Run tests
     runs-on: ubuntu-latest
     strategy:
       matrix:
         python-version: ["3.10", "3.11", "3.12", "3.13"]
-
     steps:
       - uses: actions/checkout@v5
         with:
@@ -70,7 +63,6 @@ jobs:
           working-directory: python
           sccache: true
           manylinux: auto
-
   linux:
     runs-on: ubuntu-latest
     strategy:
@@ -121,7 +113,6 @@ jobs:
             pip3 install -U pip
             pip3 install --pre "mtc_token_healing[test]" --find-links dist --force-reinstall
             pytest --import-mode=importlib
-
   windows:
     runs-on: windows-latest
     strategy:
@@ -154,7 +145,37 @@ jobs:
           set -e
           pip install --pre "mtc_token_healing[test]" --find-links dist --force-reinstall
           pytest --import-mode=importlib
-
+  macos:
+    runs-on: macos-latest
+    strategy:
+      matrix:
+        target: [x86_64, aarch64]
+    steps:
+      - uses: actions/checkout@v5
+        with:
+          fetch-depth: 0
+      - uses: actions/setup-python@v6
+        with:
+          python-version: "3.10"
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          working-directory: python
+          target: ${{ matrix.target }}
+          args: --release --out dist --interpreter 3.10
+          sccache: true
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-macos-${{ matrix.target }}
+          path: dist
+      - name: pytest
+        if: ${{ !startsWith(matrix.target, 'aarch64') }}
+        shell: bash
+        run: |
+          set -e
+          pip install --pre "mtc_token_healing[test]" --find-links dist --force-reinstall
+          pytest --import-mode=importlib
   sdist:
     needs: [test]
     runs-on: ubuntu-latest
@@ -173,7 +194,6 @@ jobs:
         with:
           name: wheels-sdist
           path: python/dist
-
   release:
     name: Release
     runs-on: ubuntu-latest

Cargo.toml

@@ -2,7 +2,7 @@
 members = ["python"]
 
 [workspace.package]
-version = "0.3.0"
+version = "0.3.1"
 edition = "2024"
 license = "MIT OR Apache-2.0"
 description = "Token healing implementation"
@@ -12,7 +12,7 @@ documentation = "https://docs.rs/mtc-token-healing"
 authors = ["Chielo Newctle <[email protected]>"]
 
 [workspace.dependencies]
-general-sam = { version = "1.0.1", features = ["trie"] }
+general-sam = { version = "1.0.3", features = ["trie"] }
 
 [package]
 name = "mtc-token-healing"

python/Cargo.toml

@@ -15,8 +15,13 @@ name = "mtc_token_healing"
 crate-type = ["cdylib"]
 
 [dependencies]
-mtc-token-healing = { version = "0.3.0", path = "..", features = ["pyo3"] }
+derive_more = { version = "2.0.1", features = ["from", "into"] }
 general-sam = { workspace = true }
-pyo3 = { version = "0.26.0", features = ["extension-module", "generate-import-lib", "abi3-py310"] }
 itertools = "0.14.0"
-derive_more = { version = "2.0.1", features = ["from", "into"] }
+mtc-token-healing = { version = "0.3.0", path = "..", features = ["pyo3"] }
+pyo3 = { version = "0.26.0", features = ["extension-module", "generate-import-lib", "abi3-py310"] }
+
+[profile.release]
+lto = true
+strip = true
+opt-level = "z"

python/pyproject.toml

@@ -13,6 +13,7 @@ classifiers = [
 dynamic = ["version"]
 
 [tool.maturin]
+features = ["pyo3/extension-module"]
 
 [project.optional-dependencies]
 test = ["pytest"]

python/src/prefix_dfs.rs

@@ -142,11 +142,11 @@ fn dfs_token_seq_trie(inputs: Vec<TokenSeqInput>) -> Vec<TokenSeqTrieNode> {
                     });
                 }
                 TravelEvent::Pop(node, _) => {
-                    if let Some(id) = rank[node.node_id] {
-                        if let Some(parent) = node.get_node().and_then(|n| rank[n.get_parent()]) {
-                            dfs_order[parent].subtree_upper = dfs_order[id].subtree_upper;
-                            dfs_order[parent].num_children += 1;
-                        }
+                    if let Some(id) = rank[node.node_id]
+                        && let Some(parent) = node.get_node().and_then(|n| rank[n.get_parent()])
+                    {
+                        dfs_order[parent].subtree_upper = dfs_order[id].subtree_upper;
+                        dfs_order[parent].num_children += 1;
                     }
                 }
             }