Fix benchmark; bump lint

rapidsai · Nov 5, 2024 · c500985 · c500985
1 parent 824e8c2
commit c500985
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/benchmarks/nx-cugraph/pytest-based/bench_algos.py b/benchmarks/nx-cugraph/pytest-based/bench_algos.py
@@ -16,6 +16,8 @@
 import networkx as nx
 import pandas as pd
 import pytest
+from collections.abc import Mapping
+
 from cugraph import datasets
 import nx_cugraph as nxcg
 
@@ -520,7 +522,7 @@ def bench_shortest_path(benchmark, graph_obj, backend_wrapper):
         iterations=iterations,
         warmup_rounds=warmup_rounds,
     )
-    assert type(result) is dict
+    assert isinstance(result, Mapping)  # dict in nx, but we duck-type
 
 
 def bench_single_source_shortest_path_length(benchmark, graph_obj, backend_wrapper):

diff --git a/python/nx-cugraph/lint.yaml b/python/nx-cugraph/lint.yaml
@@ -52,7 +52,7 @@ repos:
       - id: black
       # - id: black-jupyter
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.7.1
+    rev: v0.7.2
     hooks:
       - id: ruff
         args: [--fix-only, --show-fixes]  # --unsafe-fixes]
@@ -64,7 +64,7 @@ repos:
         additional_dependencies: &flake8_dependencies
           # These versions need updated manually
           - flake8==7.1.1
-          - flake8-bugbear==24.8.19
+          - flake8-bugbear==24.10.31
           - flake8-simplify==0.21.0
   - repo: https://github.com/asottile/yesqa
     rev: v1.5.0
@@ -80,7 +80,7 @@ repos:
         files: python/nx-cugraph/
         args: [-L, "coo,numer"]
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.7.1
+    rev: v0.7.2
     hooks:
       - id: ruff
   - repo: https://github.com/pre-commit/pre-commit-hooks