Skip to content

Commit 79a99d7

Browse files
committed
fix: WellPlatePlan.__iter__ return annotation
1 parent caf04a4 commit 79a99d7

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/useq/_plate.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
TYPE_CHECKING,
66
Any,
77
Iterable,
8+
Iterator,
89
List,
910
Sequence,
1011
Tuple,
@@ -22,7 +23,6 @@
2223
field_validator,
2324
model_validator,
2425
)
25-
from typing_extensions import Annotated
2626

2727
from useq._base_model import FrozenModel, UseqModel
2828
from useq._grid import RandomPoints, RelativeMultiPointPlan, Shape
@@ -31,6 +31,7 @@
3131

3232
if TYPE_CHECKING:
3333
from pydantic_core import core_schema
34+
from typing_extensions import Annotated
3435

3536
Index = Union[int, List[int], slice]
3637
IndexExpression = Union[Tuple[Index, ...], Index]
@@ -261,7 +262,7 @@ def rotation_matrix(self) -> np.ndarray:
261262
rads = np.radians(self.rotation)
262263
return np.array([[np.cos(rads), np.sin(rads)], [-np.sin(rads), np.cos(rads)]])
263264

264-
def __iter__(self) -> Iterable[Position]: # type: ignore
265+
def __iter__(self) -> Iterator[Position]: # type: ignore
265266
"""Iterate over the selected positions."""
266267
yield from self.image_positions
267268

0 commit comments

Comments
 (0)