diff --git a/cirq-core/cirq/work/sampler.py b/cirq-core/cirq/work/sampler.py index 59629f2d2f0..03ff9259454 100644 --- a/cirq-core/cirq/work/sampler.py +++ b/cirq-core/cirq/work/sampler.py @@ -24,6 +24,7 @@ Optional, Sequence, Tuple, + TypeVar, TYPE_CHECKING, Union, ) @@ -43,6 +44,8 @@ if TYPE_CHECKING: import cirq +T = TypeVar('T') + class Sampler(metaclass=value.ABCMetaImplementAnyOneOf): """Something capable of sampling quantum circuits. Simulator or hardware.""" @@ -486,6 +489,6 @@ def _get_measurement_shapes( return {k: (num_instances[k], qid_shape) for k, qid_shape in qid_shapes.items()} -def _chunked(iterable: Sequence[Any], n: int) -> Iterator[tuple[Any, ...]]: # pragma: no cover - it = iter(iterable) # pragma: no cover - return iter(lambda: tuple(islice(it, n)), ()) # pragma: no cover +def _chunked(iterable: Sequence[T], n: int) -> Iterator[tuple[T, ...]]: + it = iter(iterable) + return iter(lambda: tuple(islice(it, n)), ())