diff --git a/benchmarks/nx-cugraph/pytest-based/bench_algos.py b/benchmarks/nx-cugraph/pytest-based/bench_algos.py index 8852ed2a87..0c1a344234 100644 --- 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 index e6b0a150d5..823b35dfe5 100644 --- 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