arrowspace 0.13.3__tar.gz → 0.15.0__tar.gz

arrowspace-0.15.0/.github/workflows/python-publish.yml

@@ -0,0 +1,137 @@
+# .github/workflows/publish.yml
+# Publishes pyarrowspace Python package to PyPI using maturin when a release is created
+
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+  workflow_dispatch:  # Allow manual trigger for testing
+
+permissions:
+  contents: read
+
+jobs:
+  linux:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        target: [x86_64, aarch64]
+        python-version: ["3.11", "3.12", "3.13"]
+    steps:
+      - uses: actions/checkout@v4
+      
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.target }}
+          args: --release --out dist --find-interpreter
+          sccache: 'true'
+          manylinux: auto
+      
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-linux-${{ matrix.target }}-py${{ matrix.python-version }}
+          path: dist
+
+  windows:
+    runs-on: windows-latest
+    strategy:
+      matrix:
+        target: [x64]
+        python-version: ["3.11", "3.12", "3.13"]
+    steps:
+      - uses: actions/checkout@v4
+      
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          architecture: ${{ matrix.target }}
+      
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.target }}
+          args: --release --out dist --find-interpreter
+          sccache: 'true'
+      
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-windows-${{ matrix.target }}-py${{ matrix.python-version }}
+          path: dist
+
+  macos:
+    runs-on: macos-latest
+    strategy:
+      matrix:
+        target: [x86_64, aarch64]
+        python-version: ["3.11", "3.12", "3.13"]
+    steps:
+      - uses: actions/checkout@v4
+      
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.target }}
+          args: --release --out dist --find-interpreter
+          sccache: 'true'
+      
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-macos-${{ matrix.target }}-py${{ matrix.python-version }}
+          path: dist
+
+  sdist:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      
+      - name: Build sdist
+        uses: PyO3/maturin-action@v1
+        with:
+          command: sdist
+          args: --out dist
+      
+      - name: Upload sdist
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-sdist
+          path: dist
+
+  release:
+    name: Release to PyPI
+    runs-on: ubuntu-latest
+    needs: [linux, windows, macos, sdist]
+    permissions:
+      # IMPORTANT: mandatory for trusted publishing
+      id-token: write
+    environment:
+      name: pypi
+      url: https://pypi.org/project/pyarrowspace
+    steps:
+      - name: Download all artifacts
+        uses: actions/download-artifact@v4
+        with:
+          path: dist
+          pattern: wheels-*
+          merge-multiple: true
+      
+      - name: List distributions
+        run: ls -lh dist/
+      
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          skip-existing: true
+          verbose: true

{arrowspace-0.13.3 → arrowspace-0.15.0}/Cargo.lock

@@ -92,9 +92,9 @@ dependencies = [
 
 [[package]]
 name = "arrowspace"
-version = "0.13.3"
+version = "0.15.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "c6a0216c0bc0fd7f7136fb2118a9df9a517f10db14a9cd987e955a85619e0cb0"
+checksum = "05f0d5ec06ec0ff1d04461bf07210c05a72670b43e3888842cef04567b593b50"
 dependencies = [
  "approx 0.5.1",
  "dashmap",
@@ -515,7 +515,7 @@ dependencies = [
 
 [[package]]
 name = "pyarrowspace"
-version = "0.13.3"
+version = "0.15.0"
 dependencies = [
  "arrowspace",
  "numpy",

{arrowspace-0.13.3 → arrowspace-0.15.0}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "pyarrowspace"
-version = "0.13.3"
+version = "0.15.0"
 edition = "2024"
 readme = "README.md"
 
@@ -14,6 +14,6 @@ name = "arrowspace"
 crate-type = ["cdylib"]
 
 [dependencies]
-arrowspace = "0.13.3"
+arrowspace = "0.15.0"
 pyo3 = { version = "*", features = ["extension-module"] }
 numpy = "0.19"

{arrowspace-0.13.3 → arrowspace-0.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arrowspace
-Version: 0.13.3
+Version: 0.15.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{arrowspace-0.13.3 → arrowspace-0.15.0}/tests/test_0.py

@@ -24,6 +24,7 @@ aspace, gl = ArrowSpaceBuilder.build(graph_params, items)
 query1 = np.array(items[2] * 1.05, dtype=np.float64)
 hits = aspace.search(query1, gl, 1.0)  # list[(idx, score)]
 
+
 print(hits)
 assert(len(hits) == 3)
 assert(hits[0][0] == 2)
@@ -37,8 +38,8 @@ hits = aspace.search(query2, gl, 0.9)  # list[(idx, score)]
 print(hits)
 assert(len(hits) == 3)
 assert(hits[0][0] == 1)
-assert(hits[1][0] == 4)
-assert(hits[2][0] == 2)
+assert(hits[1][0] == 2)
+assert(hits[2][0] == 0)
 
 # Search comparable items (defaults: k = nitems, alpha = 0.6, beta = 0.4)
 query3 = np.array(items[2] * 1.05, dtype=np.float64)
@@ -47,8 +48,8 @@ hits = aspace.search(query3, gl, 0.6)  # list[(idx, score)]
 print(hits)
 assert(len(hits) == 3)
 assert(hits[0][0] == 1)
-assert(hits[1][0] == 4)
-assert(hits[2][0] == 0)
+assert(hits[1][0] == 3)
+assert(hits[2][0] == 2)
 
 query4 = np.array(items[2] * 1.05, dtype=np.float64)
 hits = aspace.search(query4, gl, 0.55)  # list[(idx, score)]
@@ -56,5 +57,5 @@ hits = aspace.search(query4, gl, 0.55)  # list[(idx, score)]
 print(hits)
 assert(len(hits) == 3)
 assert(hits[0][0] == 1)
-assert(hits[1][0] == 4)
-assert(hits[2][0] == 0)
+assert(hits[1][0] == 3)
+assert(hits[2][0] == 2)

{arrowspace-0.13.3 → arrowspace-0.15.0}/tests/test_2_CVE_db.py

@@ -20,8 +20,8 @@ from arrowspace import ArrowSpaceBuilder, set_debug
 
 set_debug(True)  # optional: Rust-side debug prints to stderr
 
-START_YEAR = 2020
-END_YEAR = 2020
+START_YEAR = 1999
+END_YEAR = 1999
 
 def iter_cve_json(root_dir, start=START_YEAR, end=END_YEAR):
     for path in glob.glob(os.path.join(root_dir, "**", "*.json"), recursive=True):