fastquadtree 0.4.2__tar.gz → 0.5.1__tar.gz

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/.pre-commit-config.yaml

@@ -1,12 +1,15 @@
 repos:
-  # 0) Python formatter: black
-  - repo: https://github.com/psf/black
-    rev: 24.8.0
+  # 0) Python linter and formatter: Ruff
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.6.9
     hooks:
-      - id: black
-        name: black
-        language_version: python3
-        args: ["--line-length", "88"]
+      # Lint + autofix. If Ruff fixes something, the hook exits nonzero to force a re-run.
+      - id: ruff
+        name: ruff (lint + fix)
+        args: ["--fix", "--exit-non-zero-on-fix"]
+      # Code formatter (Black replacement)
+      - id: ruff-format
+        name: ruff (format)
 
   # Local hooks that run in sequence and do not receive file args
   - repo: local
@@ -28,10 +31,9 @@ repos:
         always_run: true
 
       # 3) Python tests under coverage
-      # This both runs pytest and generates .coverage data
       - id: pytest
         name: pytest (under coverage)
         entry: pytest
         language: system
         pass_filenames: false
-        always_run: true
+        always_run: true

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/Cargo.lock

@@ -22,7 +22,7 @@ checksum = "2fd1289c04a9ea8cb22300a459a72a385d7c73d3259e2ed7dcb2af674838cfa9"
 
 [[package]]
 name = "fastquadtree"
-version = "0.4.2"
+version = "0.5.1"
 dependencies = [
  "pyo3",
 ]

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "fastquadtree"
-version = "0.4.2"
+version = "0.5.1"
 edition = "2021"
 readme = "README.md"
 
@@ -9,3 +9,9 @@ crate-type = ["rlib", "cdylib"]
 
 [dependencies]
 pyo3 = { version = "0.21", features = ["extension-module", "abi3-py38"] }
+
+[profile.release]
+opt-level = 3
+lto = "thin"
+codegen-units = 1
+panic = "abort"

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastquadtree
-Version: 0.4.2
+Version: 0.5.1
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Rust
@@ -11,6 +11,13 @@ Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Classifier: Topic :: Software Development :: Libraries
 Classifier: Typing :: Typed
 Classifier: License :: OSI Approved :: MIT License
+Requires-Dist: ruff>=0.6.0 ; extra == 'dev'
+Requires-Dist: pytest>=7.0 ; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1 ; extra == 'dev'
+Requires-Dist: coverage>=7.5 ; extra == 'dev'
+Requires-Dist: mypy>=1.10 ; extra == 'dev'
+Requires-Dist: build>=1.2.1 ; extra == 'dev'
+Provides-Extra: dev
 License-File: LICENSE
 Summary: Rust-accelerated quadtree for Python with fast inserts, range queries, and k-NN search.
 Keywords: quadtree,spatial-index,geometry,rust,pyo3,nearest-neighbor,k-nn
@@ -36,8 +43,8 @@ Project-URL: Issues, https://github.com/Elan456/fastquadtree/issues
 
 [![Rust core via PyO3](https://img.shields.io/badge/Rust-core%20via%20PyO3-orange)](https://pyo3.rs/)
 [![Built with maturin](https://img.shields.io/badge/Built%20with-maturin-1f6feb)](https://www.maturin.rs/)
-[![Code style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
-[![Type checking: mypy](https://img.shields.io/badge/type%20checking-mypy-2a6db2)](http://mypy-lang.org/)
+[![Lint and format: Ruff](https://img.shields.io/badge/Lint%20and%20format-Ruff-46a758?logo=ruff&logoColor=white)](https://docs.astral.sh/ruff/)
+
 
 
 ![Interactive_V2_Screenshot](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/interactive_v2_screenshot.png)
@@ -49,6 +56,29 @@ Rust-optimized quadtree with a simple Python API.
 - Python ≥ 3.8
 - Import path: `from fastquadtree import QuadTree`
 
+## Benchmarks
+
+fastquadtree **outperforms** all other quadtree Python packages, including the Rtree spatial index.
+
+### Library comparison
+
+![Total time](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_time.png)
+![Throughput](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_throughput.png)
+
+### Summary (largest dataset, PyQtree baseline)
+- Points: **500,000**, Queries: **500**
+--------------------
+- Fastest total: **fastquadtree** at **1.591 s**
+
+| Library | Build (s) | Query (s) | Total (s) | Speed vs PyQtree |
+|---|---:|---:|---:|---:|
+| fastquadtree | 0.165 | 1.427 | 1.591 | 5.09× |
+| Rtree        | 1.320 | 2.369 | 3.688 | 2.20× |
+| PyQtree      | 2.687 | 5.415 | 8.102 | 1.00× |
+| nontree-QuadTree | 1.284 | 9.891 | 11.175 | 0.73× |
+| quads        | 2.346 | 10.129 | 12.475 | 0.65× |
+| e-pyquadtree | 1.795 | 11.855 | 13.650 | 0.59× |
+
 ## Install
 
 ```bash
@@ -109,25 +139,7 @@ print(f"Deleted player: {deleted}")  # True
 
 You can keep the tree pure and manage your own id → object map, or let the wrapper manage it.
 
-**Option A: Manage your own map**
-
-```python
-from fastquadtree import QuadTree
-
-qt = QuadTree((0, 0, 1000, 1000), capacity=16)
-objects: dict[int, object] = {}
-
-def add(obj) -> int:
-    obj_id = qt.insert(obj.position)  # auto id
-    objects[obj_id] = obj
-    return obj_id
-
-# Later, resolve ids back to objects
-ids = [obj_id for (obj_id, x, y) in qt.query((100, 100, 300, 300))]
-selected = [objects[i] for i in ids]
-```
-
-**Option B: Ask the wrapper to track objects**
+**Wrapper Managed Objects**
 
 ```python
 from fastquadtree import QuadTree
@@ -137,7 +149,7 @@ qt = QuadTree((0, 0, 1000, 1000), capacity=16, track_objects=True)
 # Store the object alongside the point
 qt.insert((25, 40), obj={"name": "apple"})
 
-# Ask for Item objects so you can access .obj lazily
+# Ask for Item objects within a bounding box
 items = qt.query((0, 0, 100, 100), as_items=True)
 for it in items:
     print(it.id, it.x, it.y, it.obj)
@@ -151,7 +163,7 @@ qt.attach(123, my_object)  # binds object to id 123
 
 ## API
 
-### `QuadTree(bounds, capacity, *, max_depth=None, track_objects=False, start_id=1)`
+### `QuadTree(bounds, capacity, max_depth=None, track_objects=False, start_id=1)`
 
 * `bounds` — tuple `(min_x, min_y, max_x, max_y)` defines the 2D area covered by the quadtree
 * `capacity` — max number of points kept in a leaf before splitting
@@ -204,30 +216,8 @@ Full docs are in the docstrings of the [Python Shim](pysrc/fastquadtree/__init__
 * For fastest local runs, use `maturin develop --release`.
 * The wrapper keeps Python overhead low: raw tuple results by default, `Item` wrappers only when requested.
 
-## Benchmarks
-
-fastquadtree outperforms all other quadtree python packages (at least all the ones I could find and install via pip.)
-
-### Library comparison
-
-![Total time](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_time.png)
-![Throughput](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_throughput.png)
-
-### Summary (largest dataset, PyQtree baseline)
-- Points: **500,000**, Queries: **500**
-- Fastest total: **fastquadtree** at **2.207 s**
-
-| Library | Build (s) | Query (s) | Total (s) | Speed vs PyQtree |
-|---|---:|---:|---:|---:|
-| fastquadtree  | 0.321 | 1.885 | 2.207 | 4.27× |
-| Rtree        | 1.718 | 4.376 | 6.095 | 1.55× |
-| nontree-QuadTree | 1.617 | 7.643 | 9.260 | 1.02× |
-| PyQtree      | 4.349 | 5.082 | 9.431 | 1.00× |
-| quads        | 3.874 | 9.058 | 12.932 | 0.73× |
-| e-pyquadtree | 2.732 | 10.598 | 13.330 | 0.71× |
-| Brute force  | 0.019 | 19.986 | 20.005 | 0.47× |
 
-### Native vs Shim
+### Native vs Shim Benchmark
 
 **Setup**
 - Points: 500,000
@@ -277,7 +267,7 @@ Check the CLI arguments for the cross-library benchmark in `benchmarks/quadtree_
 Allowed. For k-nearest, duplicates are de-duplicated by id. For range queries you will see every inserted point.
 
 **Can I delete items from the quadtree?**
-Yes! Use `delete(id, xy)` to remove specific items. You must provide both the ID and exact location for precise deletion. This handles cases where multiple items exist at the same location. If you're using `track_objects=True`, you can also use `delete_by_object(obj, xy)` for convenient object-based deletion with O(1) lookup. The tree automatically merges nodes when item counts drop below capacity.
+Yes! Use `delete(id, xy)` to remove specific items. You must provide both the ID and exact location for precise deletion. This handles cases where multiple items exist at the same location. If you're using `track_objects=True`, you can also use `delete_by_object(obj)` for convenient object-based deletion with O(1) lookup. The tree automatically merges nodes when item counts drop below capacity.
 
 **Can I store rectangles or circles?**
 The core stores points. To index objects with extent, insert whatever representative point you choose. For rectangles you can insert centers or build an AABB tree separately.

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/README.md

@@ -13,8 +13,8 @@
 
 [![Rust core via PyO3](https://img.shields.io/badge/Rust-core%20via%20PyO3-orange)](https://pyo3.rs/)
 [![Built with maturin](https://img.shields.io/badge/Built%20with-maturin-1f6feb)](https://www.maturin.rs/)
-[![Code style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
-[![Type checking: mypy](https://img.shields.io/badge/type%20checking-mypy-2a6db2)](http://mypy-lang.org/)
+[![Lint and format: Ruff](https://img.shields.io/badge/Lint%20and%20format-Ruff-46a758?logo=ruff&logoColor=white)](https://docs.astral.sh/ruff/)
+
 
 
 ![Interactive_V2_Screenshot](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/interactive_v2_screenshot.png)
@@ -26,6 +26,29 @@ Rust-optimized quadtree with a simple Python API.
 - Python ≥ 3.8
 - Import path: `from fastquadtree import QuadTree`
 
+## Benchmarks
+
+fastquadtree **outperforms** all other quadtree Python packages, including the Rtree spatial index.
+
+### Library comparison
+
+![Total time](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_time.png)
+![Throughput](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_throughput.png)
+
+### Summary (largest dataset, PyQtree baseline)
+- Points: **500,000**, Queries: **500**
+--------------------
+- Fastest total: **fastquadtree** at **1.591 s**
+
+| Library | Build (s) | Query (s) | Total (s) | Speed vs PyQtree |
+|---|---:|---:|---:|---:|
+| fastquadtree | 0.165 | 1.427 | 1.591 | 5.09× |
+| Rtree        | 1.320 | 2.369 | 3.688 | 2.20× |
+| PyQtree      | 2.687 | 5.415 | 8.102 | 1.00× |
+| nontree-QuadTree | 1.284 | 9.891 | 11.175 | 0.73× |
+| quads        | 2.346 | 10.129 | 12.475 | 0.65× |
+| e-pyquadtree | 1.795 | 11.855 | 13.650 | 0.59× |
+
 ## Install
 
 ```bash
@@ -86,25 +109,7 @@ print(f"Deleted player: {deleted}")  # True
 
 You can keep the tree pure and manage your own id → object map, or let the wrapper manage it.
 
-**Option A: Manage your own map**
-
-```python
-from fastquadtree import QuadTree
-
-qt = QuadTree((0, 0, 1000, 1000), capacity=16)
-objects: dict[int, object] = {}
-
-def add(obj) -> int:
-    obj_id = qt.insert(obj.position)  # auto id
-    objects[obj_id] = obj
-    return obj_id
-
-# Later, resolve ids back to objects
-ids = [obj_id for (obj_id, x, y) in qt.query((100, 100, 300, 300))]
-selected = [objects[i] for i in ids]
-```
-
-**Option B: Ask the wrapper to track objects**
+**Wrapper Managed Objects**
 
 ```python
 from fastquadtree import QuadTree
@@ -114,7 +119,7 @@ qt = QuadTree((0, 0, 1000, 1000), capacity=16, track_objects=True)
 # Store the object alongside the point
 qt.insert((25, 40), obj={"name": "apple"})
 
-# Ask for Item objects so you can access .obj lazily
+# Ask for Item objects within a bounding box
 items = qt.query((0, 0, 100, 100), as_items=True)
 for it in items:
     print(it.id, it.x, it.y, it.obj)
@@ -128,7 +133,7 @@ qt.attach(123, my_object)  # binds object to id 123
 
 ## API
 
-### `QuadTree(bounds, capacity, *, max_depth=None, track_objects=False, start_id=1)`
+### `QuadTree(bounds, capacity, max_depth=None, track_objects=False, start_id=1)`
 
 * `bounds` — tuple `(min_x, min_y, max_x, max_y)` defines the 2D area covered by the quadtree
 * `capacity` — max number of points kept in a leaf before splitting
@@ -181,30 +186,8 @@ Full docs are in the docstrings of the [Python Shim](pysrc/fastquadtree/__init__
 * For fastest local runs, use `maturin develop --release`.
 * The wrapper keeps Python overhead low: raw tuple results by default, `Item` wrappers only when requested.
 
-## Benchmarks
-
-fastquadtree outperforms all other quadtree python packages (at least all the ones I could find and install via pip.)
-
-### Library comparison
-
-![Total time](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_time.png)
-![Throughput](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_throughput.png)
-
-### Summary (largest dataset, PyQtree baseline)
-- Points: **500,000**, Queries: **500**
-- Fastest total: **fastquadtree** at **2.207 s**
-
-| Library | Build (s) | Query (s) | Total (s) | Speed vs PyQtree |
-|---|---:|---:|---:|---:|
-| fastquadtree  | 0.321 | 1.885 | 2.207 | 4.27× |
-| Rtree        | 1.718 | 4.376 | 6.095 | 1.55× |
-| nontree-QuadTree | 1.617 | 7.643 | 9.260 | 1.02× |
-| PyQtree      | 4.349 | 5.082 | 9.431 | 1.00× |
-| quads        | 3.874 | 9.058 | 12.932 | 0.73× |
-| e-pyquadtree | 2.732 | 10.598 | 13.330 | 0.71× |
-| Brute force  | 0.019 | 19.986 | 20.005 | 0.47× |
 
-### Native vs Shim
+### Native vs Shim Benchmark
 
 **Setup**
 - Points: 500,000
@@ -254,7 +237,7 @@ Check the CLI arguments for the cross-library benchmark in `benchmarks/quadtree_
 Allowed. For k-nearest, duplicates are de-duplicated by id. For range queries you will see every inserted point.
 
 **Can I delete items from the quadtree?**
-Yes! Use `delete(id, xy)` to remove specific items. You must provide both the ID and exact location for precise deletion. This handles cases where multiple items exist at the same location. If you're using `track_objects=True`, you can also use `delete_by_object(obj, xy)` for convenient object-based deletion with O(1) lookup. The tree automatically merges nodes when item counts drop below capacity.
+Yes! Use `delete(id, xy)` to remove specific items. You must provide both the ID and exact location for precise deletion. This handles cases where multiple items exist at the same location. If you're using `track_objects=True`, you can also use `delete_by_object(obj)` for convenient object-based deletion with O(1) lookup. The tree automatically merges nodes when item counts drop below capacity.
 
 **Can I store rectangles or circles?**
 The core stores points. To index objects with extent, insert whatever representative point you choose. For rectangles you can insert centers or build an AABB tree separately.

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/benchmarks/benchmark_native_vs_shim.py

@@ -6,11 +6,11 @@ import gc
 import random
 import statistics as stats
 from time import perf_counter as now
+
 from tqdm import tqdm
 
-from fastquadtree._native import QuadTree as NativeQuadTree
 from fastquadtree import QuadTree as ShimQuadTree
-
+from fastquadtree._native import QuadTree as NativeQuadTree
 
 BOUNDS = (0.0, 0.0, 1000.0, 1000.0)
 CAPACITY = 20
@@ -68,9 +68,10 @@ def bench_shim(points, queries, *, track_objects: bool, with_objs: bool):
     return t_build, t_query
 
 
-def median_times(fn, points, queries, repeats: int):
+def median_times(fn, points, queries, repeats: int, desc: str = "Running"):
+    """Run benchmark multiple times and return median times."""
     builds, queries_t = [], []
-    for _ in tqdm(range(repeats)):
+    for _ in tqdm(range(repeats), desc=desc, unit="run"):
         gc.disable()
         b, q = fn(points, queries)
         gc.enable()
@@ -86,29 +87,47 @@ def main():
     ap.add_argument("--repeats", type=int, default=5)
     args = ap.parse_args()
 
+    print("Native vs Shim Benchmark")
+    print("=" * 50)
+    print("Configuration:")
+    print(f"  Points: {args.points:,}")
+    print(f"  Queries: {args.queries}")
+    print(f"  Repeats: {args.repeats}")
+    print()
+
     rng = random.Random(SEED)
     points = gen_points(args.points, rng)
     queries = gen_queries(args.queries, rng)
 
     # Warmup to load modules
+    print("Warming up...")
     _ = bench_native(points[:1000], queries[:50])
     _ = bench_shim(points[:1000], queries[:50], track_objects=False, with_objs=False)
+    print()
 
+    print("Running benchmarks...")
     n_build, n_query = median_times(
-        lambda pts, qs: bench_native(pts, qs), points, queries, args.repeats
+        lambda pts, qs: bench_native(pts, qs),
+        points,
+        queries,
+        args.repeats,
+        desc="Native",
     )
     s_build_no_map, s_query_no_map = median_times(
         lambda pts, qs: bench_shim(pts, qs, track_objects=False, with_objs=False),
         points,
         queries,
         args.repeats,
+        desc="Shim (no map)",
     )
     s_build_map, s_query_map = median_times(
         lambda pts, qs: bench_shim(pts, qs, track_objects=True, with_objs=True),
         points,
         queries,
         args.repeats,
+        desc="Shim (track+objs)",
     )
+    print()
 
     def fmt(x):
         return f"{x:.3f}"
@@ -131,7 +150,7 @@ def main():
 
 **Overhead vs Native**
 
-- No map: build {s_build_no_map / n_build:.2f}x, query {s_query_no_map / n_query:.2f}x, total {(s_build_no_map + s_query_no_map) / (n_build + n_query):.2f}x  
+- No map: build {s_build_no_map / n_build:.2f}x, query {s_query_no_map / n_query:.2f}x, total {(s_build_no_map + s_query_no_map) / (n_build + n_query):.2f}x
 - Track + objs: build {s_build_map / n_build:.2f}x, query {s_query_map / n_query:.2f}x, total {(s_build_map + s_query_map) / (n_build + n_query):.2f}x
 """
     print(md.strip())

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/benchmarks/quadtree_bench/__init__.py

@@ -6,8 +6,8 @@ implementations, including performance comparison, visualization, and analysis.
 """
 
 from .engines import Engine, get_engines
-from .runner import BenchmarkRunner, BenchmarkConfig
 from .plotting import PlotManager
+from .runner import BenchmarkConfig, BenchmarkRunner
 
 __version__ = "1.0.0"
-__all__ = ["Engine", "get_engines", "BenchmarkRunner", "BenchmarkConfig", "PlotManager"]
+__all__ = ["BenchmarkConfig", "BenchmarkRunner", "Engine", "PlotManager", "get_engines"]

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/benchmarks/quadtree_bench/engines.py

@@ -7,9 +7,11 @@ allowing fair comparison of their performance characteristics.
 
 from typing import Any, Callable, Dict, List, Optional, Tuple
 
+from pyqtree import Index as PyQTree  # Pyqtree
+
 # Built-in engines (always available in this repo)
 from pyquadtree.quadtree import QuadTree as EPyQuadTree  # e-pyquadtree
-from pyqtree import Index as PyQTree  # Pyqtree
+
 from fastquadtree import QuadTree as RustQuadTree  # fastquadtree
 
 
@@ -67,7 +69,10 @@ def _create_e_pyquadtree_engine(
             _ = qt.query(q)
 
     return Engine(
-        "e-pyquadtree", "#1f77b4", build, query  # display name  # color (blue)
+        "e-pyquadtree",
+        "#1f77b4",
+        build,
+        query,  # display name  # color (blue)
     )
 
 
@@ -105,7 +110,10 @@ def _create_fastquadtree_engine(
             _ = qt.query(q)
 
     return Engine(
-        "fastquadtree", "#ff7f0e", build, query  # display name  # color (orange)
+        "fastquadtree",
+        "#ff7f0e",
+        build,
+        query,  # display name  # color (orange)
     )
 
 
@@ -165,7 +173,10 @@ def _create_nontree_engine(
             _ = tm.get_rect((xmin, ymin, xmax - xmin, ymax - ymin))
 
     return Engine(
-        "nontree-QuadTree", "#17becf", build, query  # display name  # color (cyan)
+        "nontree-QuadTree",
+        "#17becf",
+        build,
+        query,  # display name  # color (cyan)
     )
 
 
@@ -178,7 +189,7 @@ def _create_brute_force_engine(
         # Append each item as if they were being added separately
         out = []
         for p in points:
-            out.append(p)
+            out.append(p)  # noqa: PERF402
         return out
 
     def query(points, queries):
@@ -187,7 +198,10 @@ def _create_brute_force_engine(
             _ = [p for p in points if q[0] <= p[0] <= q[2] and q[1] <= p[1] <= q[3]]
 
     return Engine(
-        "Brute force", "#9467bd", build, query  # display name  # color (purple)
+        "Brute force",
+        "#9467bd",
+        build,
+        query,  # display name  # color (purple)
     )
 
 
@@ -213,8 +227,7 @@ def _create_rtree_engine(
         # Bulk stream loading is the fastest way to build
         # Keep the same 1x1 bbox convention used elsewhere for fairness
         stream = ((i, (x, y, x + 1, y + 1), None) for i, (x, y) in enumerate(points))
-        idx = rindex.Index(stream, properties=p)
-        return idx
+        return rindex.Index(stream, properties=p)
 
     def query(idx, queries):
         # Do not materialize results into a list, just consume the generator

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/benchmarks/quadtree_bench/main.py

@@ -10,8 +10,8 @@ import argparse
 from pathlib import Path
 
 from .engines import get_engines
-from .runner import BenchmarkRunner, BenchmarkConfig
 from .plotting import PlotManager
+from .runner import BenchmarkConfig, BenchmarkRunner
 
 
 def main():

{fastquadtree-0.4.2 → fastquadtree-0.5.1}/benchmarks/quadtree_bench/plotting.py

@@ -5,7 +5,7 @@ This module handles creation of performance charts, graphs, and visualizations
 for benchmark results analysis.
 """
 
-from typing import Dict, Any, Tuple
+from typing import Any, Dict, Tuple
 
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
@@ -43,7 +43,7 @@ class PlotManager:
                         name=name,
                         legendgroup=name,
                         showlegend=show_legend,
-                        line=dict(color=color, width=3),
+                        line={"color": color, "width": 3},
                     ),
                     row=1,
                     col=col,
@@ -69,15 +69,15 @@ class PlotManager:
                 f"{self.config.n_queries} queries)"
             ),
             template="plotly_dark",
-            legend=dict(
-                orientation="v",
-                traceorder="normal",
-                xanchor="left",
-                x=0,
-                yanchor="top",
-                y=1,
-            ),
-            margin=dict(l=40, r=20, t=80, b=40),
+            legend={
+                "orientation": "v",
+                "traceorder": "normal",
+                "xanchor": "left",
+                "x": 0,
+                "yanchor": "top",
+                "y": 1,
+            },
+            margin={"l": 40, "r": 20, "t": 80, "b": 40},
             height=520,
         )
 
@@ -108,7 +108,7 @@ class PlotManager:
                         name=name,
                         legendgroup=name,
                         showlegend=False,
-                        line=dict(color=color, width=3),
+                        line={"color": color, "width": 3},
                     ),
                     row=1,
                     col=1,
@@ -122,7 +122,7 @@ class PlotManager:
                     name=name,
                     legendgroup=name,
                     showlegend=True,
-                    line=dict(color=color, width=3),
+                    line={"color": color, "width": 3},
                 ),
                 row=1,
                 col=2,
@@ -138,14 +138,14 @@ class PlotManager:
         fig.update_layout(
             title="Throughput",
             template="plotly_dark",
-            legend=dict(
-                orientation="h",
-                x=0,
-                xanchor="left",
-                y=1.08,  # above the subplots
-                yanchor="bottom",
-            ),
-            margin=dict(l=60, r=40, t=120, b=40),
+            legend={
+                "orientation": "h",
+                "x": 0,
+                "xanchor": "left",
+                "y": 1.08,  # above the subplots
+                "yanchor": "bottom",
+            },
+            margin={"l": 60, "r": 40, "t": 120, "b": 40},
             height=480,
         )
 
@@ -187,7 +187,7 @@ class PlotManager:
                 height=480,
             )
             print(f"Saved PNG images to {output_dir}/ with prefix '{output_prefix}'")
-        except Exception as e:
+        except Exception as e:  # noqa: BLE001
             print(
                 f"Failed to save PNG images. Install kaleido to enable PNG export: {e}"
             )
@@ -227,7 +227,7 @@ class PlotManager:
                         x=config.experiments,
                         y=values,
                         name=f"{label} - {engine_name}",
-                        line=dict(color=color, width=3),
+                        line={"color": color, "width": 3},
                         mode="lines+markers",
                     )
                 )
@@ -237,7 +237,7 @@ class PlotManager:
             xaxis_title="Number of points",
             yaxis_title="Time (s)" if "rate" not in metric else "Ops/sec",
             template="plotly_dark",
-            legend=dict(orientation="v"),
+            legend={"orientation": "v"},
             height=600,
         )