Skip to content

Commit b321b82

Browse files
Update walltime.py
1 parent de43347 commit b321b82

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/pytest_codspeed/instruments/walltime.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
import os
44
import warnings
5-
from collections.abc import Awaitable
65
from dataclasses import asdict, dataclass
76
from math import ceil
87
from statistics import mean, quantiles, stdev
@@ -20,6 +19,7 @@
2019
from pytest_codspeed.utils import SUPPORTS_PERF_TRAMPOLINE
2120

2221
if TYPE_CHECKING:
22+
from collections.abc import Awaitable, Iterator
2323
from typing import Any, Callable
2424

2525
from pytest import Session
@@ -188,7 +188,7 @@ def _measure_iter(
188188
marker_options: BenchmarkMarkerOptions,
189189
name: str,
190190
uri: str,
191-
) -> T:
191+
) -> Iterator[None]:
192192
benchmark_config = BenchmarkConfig.from_codspeed_config_and_marker_data(
193193
self.config, marker_options
194194
)
@@ -299,7 +299,7 @@ def _measure_pedantic_iter( # noqa: C901
299299
pedantic_options: PedanticOptions[T],
300300
name: str,
301301
uri: str,
302-
) -> T:
302+
) -> Iterator[None]:
303303
benchmark_config = BenchmarkConfig.from_codspeed_config_and_marker_data(
304304
self.config, marker_options
305305
)

0 commit comments

Comments
 (0)