sig-rust 0.1.0__tar.gz

sig_rust-0.1.0/.github/dependabot.yml

@@ -0,0 +1,28 @@
+version: 2
+updates:
+  - package-ecosystem: github-actions
+    directory: /
+    schedule:
+      interval: weekly
+    groups:
+      actions:
+        patterns: ["*"]
+    open-pull-requests-limit: 5
+
+  - package-ecosystem: cargo
+    directory: /
+    schedule:
+      interval: weekly
+    groups:
+      rust-deps:
+        patterns: ["*"]
+    open-pull-requests-limit: 5
+
+  - package-ecosystem: pip
+    directory: /
+    schedule:
+      interval: weekly
+    groups:
+      python-deps:
+        patterns: ["*"]
+    open-pull-requests-limit: 5

sig_rust-0.1.0/.github/workflows/cicd.yml

@@ -0,0 +1,213 @@
+name: CICD
+
+on:
+  pull_request:
+  push:
+    branches: [main]
+
+permissions:
+  contents: read
+
+jobs:
+  rust:
+    name: Rust (${{ matrix.toolchain }})
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        toolchain: [stable, nightly]
+    steps:
+      - name: Checkout
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          persist-credentials: false
+
+      - name: Install Rust toolchain
+        run: |
+          rustup toolchain install ${{ matrix.toolchain }} --profile minimal \
+            --component clippy,rustfmt,llvm-tools-preview
+          rustup default ${{ matrix.toolchain }}
+
+      - name: Cache cargo
+        uses: Swatinem/rust-cache@c19371144df3bb44fab255c43d04cbc2ab54d1c4  # v2.9.1
+
+      - name: Install cargo-llvm-cov
+        uses: taiki-e/install-action@80e6af7a2ec7f280fffe2d0a9d3a12a9d11d86e9  # v2.75.1
+        with:
+          tool: cargo-llvm-cov
+
+      - name: Format check
+        run: cargo fmt --check
+
+      - name: Clippy
+        run: cargo clippy --all-targets --all-features -- -D warnings
+
+      - name: Test with coverage
+        if: matrix.toolchain == 'stable'
+        run: cargo llvm-cov --lcov --output-path rust-lcov.info --ignore-filename-regex python
+
+      - name: Test (nightly)
+        if: matrix.toolchain == 'nightly'
+        run: cargo test
+
+      - name: Upload Rust coverage
+        if: matrix.toolchain == 'stable'
+        uses: actions/upload-artifact@bbbca2ddaa5d8feaa63e36b76fdaad77386f024f  # v7.0.0
+        with:
+          name: rust-coverage
+          path: rust-lcov.info
+
+  python:
+    name: Python ${{ matrix.python-version }}
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
+    steps:
+      - name: Checkout
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          persist-credentials: false
+
+      - name: Install Rust toolchain
+        run: rustup toolchain install stable --profile minimal
+
+      - name: Cache cargo
+        uses: Swatinem/rust-cache@c19371144df3bb44fab255c43d04cbc2ab54d1c4  # v2.9.1
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@cec208311dfd045dd5311c1add060b2062131d57  # v8.0.0
+        with:
+          python-version: ${{ matrix.python-version }}
+          enable-cache: true
+          cache-dependency-glob: |
+            **/pyproject.toml
+            **/uv.lock
+
+      - name: Install dependencies
+        run: uv sync --all-extras --group dev --group test
+
+      - name: Lint
+        run: uv run ruff check .
+
+      - name: Format check
+        run: uv run ruff format --check .
+
+      - name: Build extension
+        run: uv run maturin develop --release
+
+      - name: Type check
+        run: uv run ty check
+
+      - name: Run tests with coverage
+        run: uv run pytest -v --cov=python/sig_rust --cov-report=term-missing --cov-report=xml:python-coverage.xml
+
+      - name: Upload Python coverage
+        if: matrix.python-version == '3.14'
+        uses: actions/upload-artifact@bbbca2ddaa5d8feaa63e36b76fdaad77386f024f  # v7.0.0
+        with:
+          name: python-coverage
+          path: python-coverage.xml
+
+  combined-coverage:
+    name: Combined coverage
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          persist-credentials: false
+
+      - name: Install Rust toolchain
+        run: |
+          rustup toolchain install stable --profile minimal \
+            --component llvm-tools-preview
+          rustup default stable
+
+      - name: Cache cargo
+        uses: Swatinem/rust-cache@c19371144df3bb44fab255c43d04cbc2ab54d1c4  # v2.9.1
+
+      - name: Install cargo-llvm-cov
+        uses: taiki-e/install-action@80e6af7a2ec7f280fffe2d0a9d3a12a9d11d86e9  # v2.75.1
+        with:
+          tool: cargo-llvm-cov
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@cec208311dfd045dd5311c1add060b2062131d57  # v8.0.0
+        with:
+          python-version: "3.14"
+          enable-cache: true
+          cache-dependency-glob: |
+            **/pyproject.toml
+            **/uv.lock
+
+      - name: Install dependencies
+        run: uv sync --all-extras --group dev --group test
+
+      - name: Instrument and run tests
+        run: |
+          eval "$(cargo llvm-cov show-env --export-prefix)"
+          export CARGO_TARGET_DIR="$CARGO_LLVM_COV_TARGET_DIR"
+          cargo llvm-cov clean --workspace
+          cargo test
+          uv run maturin develop
+          uv run pytest --cov=python/sig_rust --cov-report=xml:python-coverage.xml
+          cargo llvm-cov report --lcov --output-path combined-lcov.info
+
+      - name: Upload combined coverage
+        uses: actions/upload-artifact@bbbca2ddaa5d8feaa63e36b76fdaad77386f024f  # v7.0.0
+        with:
+          name: combined-coverage
+          path: |
+            combined-lcov.info
+            python-coverage.xml
+
+  coverage:
+    name: Upload coverage
+    needs: [rust, python, combined-coverage]
+    runs-on: ubuntu-latest
+    if: github.event_name == 'push' && github.ref == 'refs/heads/main'
+    environment: codecov
+    permissions:
+      contents: read
+      id-token: write
+    steps:
+      - name: Checkout
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          persist-credentials: false
+
+      - name: Download Rust coverage
+        uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c  # v8.0.1
+        with:
+          name: rust-coverage
+
+      - name: Download combined coverage
+        uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c  # v8.0.1
+        with:
+          name: combined-coverage
+
+      - name: Upload Rust-only coverage to Codecov
+        uses: codecov/codecov-action@57e3a136b779b570ffcdbf80b3bdc90e7fab3de2  # v6.0.0
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          files: rust-lcov.info
+          flags: rust
+          fail_ci_if_error: true
+
+      - name: Upload Python coverage to Codecov
+        uses: codecov/codecov-action@57e3a136b779b570ffcdbf80b3bdc90e7fab3de2  # v6.0.0
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          files: python-coverage.xml
+          flags: python
+          fail_ci_if_error: true
+
+      - name: Upload combined coverage to Codecov
+        uses: codecov/codecov-action@57e3a136b779b570ffcdbf80b3bdc90e7fab3de2  # v6.0.0
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          files: combined-lcov.info
+          flags: combined
+          fail_ci_if_error: true

sig_rust-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,88 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+permissions:
+  contents: read
+
+jobs:
+  sdist:
+    name: Build sdist
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          persist-credentials: false
+
+      - name: Build sdist
+        uses: PyO3/maturin-action@04ac600d27cdf7a9a280dadf7147097c42b757ad  # v1.50.1
+        with:
+          command: sdist
+          args: --out dist
+
+      - name: Upload sdist
+        uses: actions/upload-artifact@bbbca2ddaa5d8feaa63e36b76fdaad77386f024f  # v7.0.0
+        with:
+          name: dist-sdist
+          path: dist/*.tar.gz
+
+  wheels:
+    name: Build wheels (${{ matrix.target }} / ${{ matrix.os }})
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - os: ubuntu-latest
+            target: x86_64
+          - os: ubuntu-latest
+            target: aarch64
+          - os: macos-latest
+            target: x86_64
+          - os: macos-latest
+            target: aarch64
+          - os: windows-latest
+            target: x86_64
+    steps:
+      - name: Checkout
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
+        with:
+          persist-credentials: false
+
+      - name: Build wheels
+        uses: PyO3/maturin-action@04ac600d27cdf7a9a280dadf7147097c42b757ad  # v1.50.1
+        with:
+          target: ${{ matrix.target }}
+          args: --release --out dist --find-interpreter
+          manylinux: auto
+
+      - name: Upload wheels
+        uses: actions/upload-artifact@bbbca2ddaa5d8feaa63e36b76fdaad77386f024f  # v7.0.0
+        with:
+          name: dist-${{ matrix.os }}-${{ matrix.target }}
+          path: dist/*.whl
+
+  publish:
+    name: Publish
+    needs: [sdist, wheels]
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      id-token: write
+      contents: read
+    steps:
+      - name: Download all artifacts
+        uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c  # v8.0.1
+        with:
+          pattern: dist-*
+          merge-multiple: true
+          path: dist
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@cec208311dfd045dd5311c1add060b2062131d57  # v8.0.0
+
+      - name: Publish to PyPI
+        run: uv publish dist/*

sig_rust-0.1.0/.gitignore

@@ -0,0 +1,81 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[codz]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+.python-version
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py.cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+
+# Ruff
+.ruff_cache/
+
+# mypy / type checkers
+.mypy_cache/
+.dmypy.json
+dmypy.json
+.pyre/
+.pytype/
+
+# Rust
+/target
+**/*.rs.bk
+.cargo/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+Thumbs.db
+
+# PyPI
+.pypirc
+

sig_rust-0.1.0/CLAUDE.md

@@ -0,0 +1,73 @@
+# sig-rust
+
+Rust port of sig-light: path signature and log signature computation with PyO3 bindings.
+
+## Build Commands
+
+Requires [just](https://github.com/casey/just) for task running.
+
+```bash
+just sync                  # Install all dependencies
+just build                 # Build native extension (maturin develop --release)
+just check                 # Run all checks (Rust + Python)
+just rust-check            # Rust only: fmt, clippy, test
+just python-check          # Python only: ruff, ty, pytest
+just format-all            # Auto-format everything
+just test-cov-all          # Full coverage (Rust + Python)
+just bench                 # Quick benchmark vs iisignature
+just bench-full            # Full profiling suite
+```
+
+Individual commands:
+
+```bash
+just rust-test             # cargo test
+just rust-lint             # cargo clippy
+just rust-fmt-check        # cargo fmt --check
+just rust-fmt              # cargo fmt
+just rust-test-cov         # cargo llvm-cov
+
+just build                 # maturin develop --release
+just test                  # pytest
+just lint                  # ruff check
+just format-check          # ruff format --check
+just format                # ruff format
+just typecheck             # ty check
+just test-cov              # pytest --cov
+```
+
+## Architecture
+
+- `src/lib.rs` -- crate root, module declarations
+- `src/error.rs` -- `SigError` enum (thiserror)
+- `src/types.rs` -- `Dim`, `Depth`, `LevelList`, `BatchedLevelList` newtypes
+- `src/algebra.rs` -- tensor algebra: multiply, unconcatenate, log, segment signatures, adjoints
+- `src/lyndon.rs` -- Lyndon words, factorization, projection matrices (SVD)
+- `src/signature.rs` -- `sig()`, `siglength()`, `sigcombine()`, binary tree reduction
+- `src/logsignature.rs` -- `logsig()`, `prepare()`, `logsiglength()`, BCH heuristic
+- `src/bch.rs` -- Baker-Campbell-Hausdorff compiled program: precomputes bracket structure constants and emits flat FMA instructions for fast log-signature evaluation
+- `src/backprop.rs` -- `sigbackprop()`, `sigjacobian()`, `logsigbackprop()`, reversibility trick
+- `src/rotational.rs` -- 2D rotation-invariant features
+- `src/transforms.rs` -- `sigjoin()`, `sigscale()` + backprop
+- `src/python.rs` -- PyO3 bindings with adaptive rayon parallelism
+
+## Key Design Decisions
+
+- `LevelList`: flat `Vec<f64>` with offset-based level access for cache locality. All tensor algebra operates on this type.
+- Reversibility trick in backprop: recovers prefix signatures via `tensor_unconcatenate_into` instead of storing all intermediates (O(siglength) vs O(n * siglength) memory).
+- BCH vs S method: `logsignature.rs` selects BCH for small configs (d<=3, m<=4) where compiled FMA wins; S method (Horner tensor log) for larger configs.
+- Rayon parallelism: `collect_par_or_seq` in `python.rs` uses a work-estimation heuristic (500K threshold, min 16 batch items) to avoid thread pool overhead on small batches.
+- Release profile: fat LTO + single codegen unit for maximum optimization.
+
+## CI/CD
+
+- **CI** (`.github/workflows/cicd.yml`): Rust stable+nightly (fmt, clippy, test+coverage), Python 3.10-3.14 (ruff, ty, pytest+coverage). Coverage uploaded to Codecov with `rust` and `python` flags.
+- **Publish** (`.github/workflows/publish.yml`): On GitHub release, builds wheels for Linux/macOS/Windows (x86_64 + aarch64) via maturin-action, publishes to PyPI.
+- **Dependabot**: Weekly grouped updates for GitHub Actions, Cargo, and pip.
+- Codecov token lives in the `codecov` GitHub environment. PyPI uses trusted publishing via the `pypi` environment.
+
+## Testing
+
+- Rust tests: `#[cfg(test)]` modules in `bch.rs` (bracket identities, BCH/S-method agreement, backprop agreement).
+- Python tests: `python/tests/test_python_bindings.py` (105 tests) covering all public API functions, finite-difference gradient checks, and iisignature cross-validation.
+- iisignature compatibility tests run when `iisignature` is installed (included in the `test` dependency group).