nmslib-rs 0.1.0__tar.gz

nmslib_rs-0.1.0/.cargo/config.toml

@@ -0,0 +1,3 @@
+[env]
+# Use the project venv (Python 3.12) for pyo3 builds, matching maturin.
+PYO3_PYTHON = { value = ".venv/bin/python", relative = true }

nmslib_rs-0.1.0/.github/workflows/ci.yml

@@ -0,0 +1,51 @@
+name: ci
+
+on:
+  push:
+    branches: ["**"]
+  pull_request:
+    branches: ["**"]
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  test:
+    name: test (${{ matrix.os }})
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-latest]
+    steps:
+      - uses: actions/checkout@v6
+
+      - uses: dtolnay/rust-toolchain@stable
+        with:
+          components: rustfmt
+
+      - name: Check formatting
+        run: cargo fmt --check
+
+      - uses: actions/setup-python@v6
+        with:
+          python-version: "3.12"
+
+      # .cargo/config.toml points PyO3 at .venv/bin/python, so the venv
+      # has to live at that path for cargo test to link.
+      - name: Create virtual environment
+        run: |
+          python -m venv .venv
+          .venv/bin/pip install --upgrade pip maturin numpy pytest
+
+      - name: Rust tests
+        run: cargo test
+
+      - name: Build and install extension
+        run: |
+          source .venv/bin/activate
+          maturin develop --release
+
+      - name: Python tests
+        run: .venv/bin/pytest tests

nmslib_rs-0.1.0/.github/workflows/release.yml

@@ -0,0 +1,115 @@
+name: release
+
+on:
+  push:
+    tags:
+      - "v*"
+  # Manual runs build and test every wheel without publishing.
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  check-version:
+    name: check-version
+    runs-on: ubuntu-latest
+    if: startsWith(github.ref, 'refs/tags/')
+    steps:
+      - uses: actions/checkout@v6
+      - name: Tag matches Cargo.toml version
+        run: |
+          version=$(sed -n 's/^version = "\(.*\)"/\1/p' Cargo.toml | head -1)
+          if [ "v$version" != "$GITHUB_REF_NAME" ]; then
+            echo "Tag $GITHUB_REF_NAME does not match Cargo.toml version $version" >&2
+            exit 1
+          fi
+
+  wheels:
+    name: wheels (${{ matrix.platform.name }})
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      fail-fast: false
+      matrix:
+        platform:
+          - { name: linux-x86_64, runner: ubuntu-latest, target: x86_64, manylinux: auto }
+          - { name: linux-aarch64, runner: ubuntu-24.04-arm, target: aarch64, manylinux: auto }
+          - { name: macos-x86_64, runner: macos-15-intel, target: x86_64 }
+          - { name: macos-aarch64, runner: macos-latest, target: aarch64 }
+          - { name: windows-x64, runner: windows-latest, target: x64 }
+    steps:
+      - uses: actions/checkout@v6
+
+      # --find-interpreter builds one wheel per Python it finds. The manylinux
+      # container ships its own interpreters; macOS and Windows use these.
+      - uses: actions/setup-python@v6
+        with:
+          python-version: |
+            3.12
+            3.13
+            3.14
+
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          manylinux: ${{ matrix.platform.manylinux }}
+          args: --release --out dist --find-interpreter
+
+      - uses: astral-sh/setup-uv@v7
+
+      - name: Test wheel
+        shell: bash
+        run: |
+          set -e
+          uv venv --python 3.12 .venv-test
+          if [ -d .venv-test/Scripts ]; then source .venv-test/Scripts/activate; else source .venv-test/bin/activate; fi
+          uv pip install nmslib-rs --no-index --no-deps --find-links dist
+          uv pip install numpy pytest
+          pytest tests
+
+      - uses: actions/upload-artifact@v6
+        with:
+          name: wheels-${{ matrix.platform.name }}
+          path: dist
+
+  sdist:
+    name: sdist
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - name: Build sdist
+        uses: PyO3/maturin-action@v1
+        with:
+          command: sdist
+          args: --out dist
+      - uses: actions/upload-artifact@v6
+        with:
+          name: wheels-sdist
+          path: dist
+
+  publish:
+    name: publish
+    runs-on: ubuntu-latest
+    needs: [check-version, wheels, sdist]
+    if: startsWith(github.ref, 'refs/tags/')
+    environment: pypi
+    permissions:
+      contents: write
+      id-token: write
+    steps:
+      - uses: actions/download-artifact@v7
+        with:
+          pattern: wheels-*
+          path: dist
+          merge-multiple: true
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          packages-dir: dist
+
+      - name: Create GitHub release
+        env:
+          GH_TOKEN: ${{ github.token }}
+        run: gh release create "$GITHUB_REF_NAME" dist/* --repo "$GITHUB_REPOSITORY" --title "$GITHUB_REF_NAME" --generate-notes

nmslib_rs-0.1.0/.gitignore

@@ -0,0 +1,5 @@
+target/
+.venv/
+bench/data/
+__pycache__/
+.venv-nmslib-native/