Update iterate type hint

Farama-Foundation · Sep 20, 2024 · 148c212 · 148c212
1 parent 4bc80cf
commit 148c212
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gymnasium/vector/utils/space_utils.py b/gymnasium/vector/utils/space_utils.py
@@ -271,7 +271,7 @@ def _batch_spaces_undefined(spaces: list[Graph | Text | Sequence | OneOf]):
 
 
 @singledispatch
-def iterate(space: Space[T_cov], items: Iterable[T_cov]) -> Iterator:
+def iterate(space: Space[T_cov], items: T_cov) -> Iterator:
     """Iterate over the elements of a (batched) space.
 
     Args: