fastquadtree 0.5.0__tar.gz → 0.6.0__tar.gz

{fastquadtree-0.5.0 → fastquadtree-0.6.0}/.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.5.0 → fastquadtree-0.6.0}/Cargo.lock

@@ -22,9 +22,10 @@ checksum = "2fd1289c04a9ea8cb22300a459a72a385d7c73d3259e2ed7dcb2af674838cfa9"
 
 [[package]]
 name = "fastquadtree"
-version = "0.5.0"
+version = "0.6.0"
 dependencies = [
  "pyo3",
+ "smallvec",
 ]
 
 [[package]]

{fastquadtree-0.5.0 → fastquadtree-0.6.0}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "fastquadtree"
-version = "0.5.0"
+version = "0.6.0"
 edition = "2021"
 readme = "README.md"
 
@@ -9,6 +9,7 @@ crate-type = ["rlib", "cdylib"]
 
 [dependencies]
 pyo3 = { version = "0.21", features = ["extension-module", "abi3-py38"] }
+smallvec = "1.15.1"
 
 [profile.release]
 opt-level = 3

{fastquadtree-0.5.0 → fastquadtree-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastquadtree
-Version: 0.5.0
+Version: 0.6.0
 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
@@ -28,15 +35,15 @@ Project-URL: Issues, https://github.com/Elan456/fastquadtree/issues
 [![Wheels](https://img.shields.io/pypi/wheel/fastquadtree.svg)](https://pypi.org/project/fastquadtree/#files)
 [![License: MIT](https://img.shields.io/pypi/l/fastquadtree.svg)](LICENSE)
 
-[![Downloads total](https://static.pepy.tech/badge/fastquadtree)](https://pepy.tech/projects/fastquadtree)
-[![Downloads month](https://static.pepy.tech/badge/fastquadtree/month)](https://pepy.tech/projects/fastquadtree)
+[![PyPI Downloads](https://static.pepy.tech/personalized-badge/fastquadtree?period=total&units=INTERNATIONAL_SYSTEM&left_color=GRAY&right_color=BLUE&left_text=Total+Downloads)](https://pepy.tech/projects/fastquadtree)
 
 [![Build](https://github.com/Elan456/fastquadtree/actions/workflows/release.yml/badge.svg)](https://github.com/Elan456/fastquadtree/actions/workflows/ci.yml)
 [![Codecov](https://codecov.io/gh/Elan456/fastquadtree/branch/main/graph/badge.svg)](https://codecov.io/gh/Elan456/fastquadtree)
 
 [![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)
+[![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+
 
 
 ![Interactive_V2_Screenshot](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/interactive_v2_screenshot.png)
@@ -58,18 +65,27 @@ fastquadtree **outperforms** all other quadtree Python packages, including the R
 ![Throughput](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_throughput.png)
 
 ### Summary (largest dataset, PyQtree baseline)
-- Points: **500,000**, Queries: **500**
+- Points: **250,000**, Queries: **500**
 --------------------
-- Fastest total: **fastquadtree** at **1.591 s**
+- Fastest total: **fastquadtree** at **0.120 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× |
+| fastquadtree | 0.031 | 0.089 | 0.120 | 14.64× |
+| Shapely STRtree | 0.179 | 0.100 | 0.279 | 6.29× |
+| nontree-QuadTree | 0.595 | 0.605 | 1.200 | 1.46× |
+| Rtree        | 0.961 | 0.300 | 1.261 | 1.39× |
+| e-pyquadtree | 1.005 | 0.660 | 1.665 | 1.05× |
+| PyQtree      | 1.492 | 0.263 | 1.755 | 1.00× |
+| quads        | 1.407 | 0.484 | 1.890 | 0.93× |
+
+#### Benchmark Configuration
+| Parameter | Value |
+|---|---:|
+| Bounds | (0, 0, 1000, 1000) |
+| Max points per node | 128 |
+| Max depth | 16 |
+| Queries per experiment | 500 |
 
 ## Install
 
@@ -273,7 +289,7 @@ MIT. See `LICENSE`.
 
 ## Acknowledgments
 
-* Python libraries compared: [PyQtree], [e-pyquadtree], [Rtree], [nontree], [quads]
+* Python libraries compared: [PyQtree], [e-pyquadtree], [Rtree], [nontree], [quads], [Shapely]
 * Built with [PyO3] and [maturin]
 
 [PyQtree]: https://pypi.org/project/pyqtree/
@@ -283,3 +299,5 @@ MIT. See `LICENSE`.
 [Rtree]: https://pypi.org/project/Rtree/
 [nontree]: https://pypi.org/project/nontree/
 [quads]: https://pypi.org/project/quads/
+[Shapely]: https://pypi.org/project/Shapely/
+

{fastquadtree-0.5.0 → fastquadtree-0.6.0}/README.md

@@ -5,15 +5,15 @@
 [![Wheels](https://img.shields.io/pypi/wheel/fastquadtree.svg)](https://pypi.org/project/fastquadtree/#files)
 [![License: MIT](https://img.shields.io/pypi/l/fastquadtree.svg)](LICENSE)
 
-[![Downloads total](https://static.pepy.tech/badge/fastquadtree)](https://pepy.tech/projects/fastquadtree)
-[![Downloads month](https://static.pepy.tech/badge/fastquadtree/month)](https://pepy.tech/projects/fastquadtree)
+[![PyPI Downloads](https://static.pepy.tech/personalized-badge/fastquadtree?period=total&units=INTERNATIONAL_SYSTEM&left_color=GRAY&right_color=BLUE&left_text=Total+Downloads)](https://pepy.tech/projects/fastquadtree)
 
 [![Build](https://github.com/Elan456/fastquadtree/actions/workflows/release.yml/badge.svg)](https://github.com/Elan456/fastquadtree/actions/workflows/ci.yml)
 [![Codecov](https://codecov.io/gh/Elan456/fastquadtree/branch/main/graph/badge.svg)](https://codecov.io/gh/Elan456/fastquadtree)
 
 [![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)
+[![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+
 
 
 ![Interactive_V2_Screenshot](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/interactive_v2_screenshot.png)
@@ -35,18 +35,27 @@ fastquadtree **outperforms** all other quadtree Python packages, including the R
 ![Throughput](https://raw.githubusercontent.com/Elan456/fastquadtree/main/assets/quadtree_bench_throughput.png)
 
 ### Summary (largest dataset, PyQtree baseline)
-- Points: **500,000**, Queries: **500**
+- Points: **250,000**, Queries: **500**
 --------------------
-- Fastest total: **fastquadtree** at **1.591 s**
+- Fastest total: **fastquadtree** at **0.120 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× |
+| fastquadtree | 0.031 | 0.089 | 0.120 | 14.64× |
+| Shapely STRtree | 0.179 | 0.100 | 0.279 | 6.29× |
+| nontree-QuadTree | 0.595 | 0.605 | 1.200 | 1.46× |
+| Rtree        | 0.961 | 0.300 | 1.261 | 1.39× |
+| e-pyquadtree | 1.005 | 0.660 | 1.665 | 1.05× |
+| PyQtree      | 1.492 | 0.263 | 1.755 | 1.00× |
+| quads        | 1.407 | 0.484 | 1.890 | 0.93× |
+
+#### Benchmark Configuration
+| Parameter | Value |
+|---|---:|
+| Bounds | (0, 0, 1000, 1000) |
+| Max points per node | 128 |
+| Max depth | 16 |
+| Queries per experiment | 500 |
 
 ## Install
 
@@ -250,7 +259,7 @@ MIT. See `LICENSE`.
 
 ## Acknowledgments
 
-* Python libraries compared: [PyQtree], [e-pyquadtree], [Rtree], [nontree], [quads]
+* Python libraries compared: [PyQtree], [e-pyquadtree], [Rtree], [nontree], [quads], [Shapely]
 * Built with [PyO3] and [maturin]
 
 [PyQtree]: https://pypi.org/project/pyqtree/
@@ -259,4 +268,5 @@ MIT. See `LICENSE`.
 [maturin]: https://www.maturin.rs/
 [Rtree]: https://pypi.org/project/Rtree/
 [nontree]: https://pypi.org/project/nontree/
-[quads]: https://pypi.org/project/quads/
+[quads]: https://pypi.org/project/quads/
+[Shapely]: https://pypi.org/project/Shapely/

{fastquadtree-0.5.0 → fastquadtree-0.6.0}/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.5.0 → fastquadtree-0.6.0}/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.5.0 → fastquadtree-0.6.0}/benchmarks/quadtree_bench/engines.py

@@ -7,9 +7,14 @@ allowing fair comparison of their performance characteristics.
 
 from typing import Any, Callable, Dict, List, Optional, Tuple
 
+import numpy as np
+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 shapely import box as shp_box, points  # Shapely 2.x
+from shapely.strtree import STRtree
+
 from fastquadtree import QuadTree as RustQuadTree  # fastquadtree
 
 
@@ -67,7 +72,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)
     )
 
 
@@ -96,8 +104,7 @@ def _create_fastquadtree_engine(
 
     def build(points):
         qt = RustQuadTree(bounds, max_points, max_depth=max_depth)
-        for p in points:
-            qt.insert(p)
+        qt.insert_many_points(points)
         return qt
 
     def query(qt, queries):
@@ -105,7 +112,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 +175,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 +191,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 +200,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 +229,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
@@ -226,6 +241,40 @@ def _create_rtree_engine(
     return Engine("Rtree", "#e377c2", build, query)
 
 
+def _create_strtree_engine(
+    bounds: Tuple[int, int, int, int], max_points: int, max_depth: int
+) -> Optional[Engine]:
+    """Create engine adapter for Shapely STRtree (optional)."""
+
+    def build(points_list: List[Tuple[int, int]]):
+        # Build geometries efficiently
+
+        xs = np.fromiter(
+            (x for x, _ in points_list), dtype="float32", count=len(points_list)
+        )
+        ys = np.fromiter(
+            (y for _, y in points_list), dtype="float32", count=len(points_list)
+        )
+        geoms = points(xs, ys)  # vectorized Point creation
+        assert type(geoms) is np.ndarray
+        tree = STRtree(geoms, node_capacity=max_points)
+        # Keep geoms alive next to the tree
+        return (tree, geoms)
+
+    def query(built, queries: List[Tuple[int, int, int, int]]):
+        tree, _geoms = built
+        for xmin, ymin, xmax, ymax in queries:
+            window = shp_box(xmin, ymin, xmax, ymax)
+            # Shapely 2.x returns ndarray of indices for a single geometry
+            res = tree.query(window)
+            # Consume results without materializing to keep parity with other engines
+            if hasattr(res, "__iter__"):
+                for _ in res:
+                    pass
+
+    return Engine("Shapely STRtree", "#7f7f7f", build, query)
+
+
 def get_engines(
     bounds: Tuple[int, int, int, int] = (0, 0, 1000, 1000),
     max_points: int = 20,
@@ -255,6 +304,7 @@ def get_engines(
         _create_quads_engine,
         _create_nontree_engine,
         _create_rtree_engine,
+        _create_strtree_engine,
     ]
 
     for engine_creator in optional_engines:

{fastquadtree-0.5.0 → fastquadtree-0.6.0}/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():
@@ -25,10 +25,10 @@ def main():
     parser.add_argument(
         "--max-points",
         type=int,
-        default=20,
+        default=128,
         help="Maximum points per node before splitting",
     )
-    parser.add_argument("--max-depth", type=int, default=10, help="Maximum tree depth")
+    parser.add_argument("--max-depth", type=int, default=16, help="Maximum tree depth")
     parser.add_argument(
         "--n-queries", type=int, default=500, help="Number of queries per experiment"
     )
@@ -41,7 +41,7 @@ def main():
     parser.add_argument(
         "--max-experiment-points",
         type=int,
-        default=500_000,
+        default=250_000,
         help="Maximum number of points in largest experiment",
     )
     parser.add_argument(

{fastquadtree-0.5.0 → fastquadtree-0.6.0}/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,
         )