Add support for (naive) batching

gymnasium/vector/utils/space_utils.py

@@ -23,6 +23,7 @@
     GraphInstance,
     MultiBinary,
     MultiDiscrete,
+    OneOf,
     Sequence,
     Space,
     Text,
@@ -121,6 +122,11 @@ def _batch_space_dict(space: Dict, n: int = 1):
     )
 
 
+@batch_space.register(OneOf)
+def _batch_space_oneof(space: OneOf, n: int = 1):
+    return Sequence(space)
+
+
 @batch_space.register(Graph)
 @batch_space.register(Text)
 @batch_space.register(Sequence)
@@ -227,6 +233,11 @@ def _iterate_dict(space: Dict, items: dict[str, Any]):
         yield OrderedDict({key: value for key, value in zip(keys, item)})
 
 
+@iterate.register(Sequence)
+def _iterate_sequence(space: Sequence, items: list[Any]):
+    yield from items
+
+
 @singledispatch
 def concatenate(
     space: Space, items: Iterable, out: tuple[Any, ...] | dict[str, Any] | np.ndarray