Allow sequence to accept stacked np arrays if the feature space is Box

gymnasium/spaces/sequence.py

@@ -119,9 +119,13 @@ def sample(
 
     def contains(self, x: Any) -> bool:
         """Return boolean specifying if x is a valid member of this space."""
-        return isinstance(x, collections.abc.Sequence) and all(
-            self.feature_space.contains(item) for item in x
-        )
+        # by definition, any sequence is an iterable
+        if not isinstance(x, collections.abc.Iterable):
+            return False
+        if not all(self.feature_space.contains(item) for item in x):
+            return False
+
+        return True
 
     def __repr__(self) -> str:
         """Gives a string representation of this space."""

tests/spaces/test_sequence.py

@@ -6,6 +6,15 @@
 import gymnasium as gym
 
 
+def test_stacked_box():
+    """Tests that sequence with a feature space of Box allows stacked np arrays."""
+    space = gym.spaces.Sequence(gym.spaces.Box(0, 1, shape=(3,)))
+    sample = np.float32(np.random.rand(5, 3))
+    assert space.contains(
+        sample
+    ), "Something went wrong, should be able to accept stacked np arrays for Box feature space."
+
+
 def test_sample():
     """Tests the sequence sampling works as expects and the errors are correctly raised."""
     space = gym.spaces.Sequence(gym.spaces.Box(0, 1))