Skip to content

Commit

Permalink
fix: changing callable to typing.Callable
Browse files Browse the repository at this point in the history
  • Loading branch information
adamamer20 committed Oct 1, 2024
1 parent 2dedbec commit ef664ae
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions examples/sugarscape_ig/performance_comparison.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
AntPolarsNumbaParallel,
)
from ss_polars.model import SugarscapePolars

from typing_extensions import Callable

class SugarScapeSetup:
def __init__(self, n: int):
Expand Down Expand Up @@ -158,11 +158,11 @@ def mesa_frames_polars_numba_parallel(setup: SugarScapeSetup):

def plot_and_print_benchmark(
labels: list[str],
kernels: list[callable],
kernels: list[Callable],
n_range: list[int],
title: str,
image_path: str,
equality_check: callable | None = None,
equality_check: Callable | None = None,
):
out = perfplot.bench(
setup=SugarScapeSetup,
Expand Down

0 comments on commit ef664ae

Please sign in to comment.