gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-17 20:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Stack to Sequence space - #217 (#284)

Browse Source 
Co-authored-by: StringTheory <mark.m.towers@gmail.com>
This commit is contained in:
Kevlyn Kadamala
2023-01-30 18:27:32 +05:30
committed by GitHub
parent 33b9884ab6
commit 8031da71e6
8 changed files with 111 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/spaces/test_utils.py
View File

@@ -5,7 +5,7 @@ import numpy as np
import pytest
import gymnasium as gym
from gymnasium.spaces import Box, Graph, utils
from gymnasium.spaces import Box, Graph, Sequence, utils
from gymnasium.utils.env_checker import data_equivalence
from tests.spaces.utils import TESTING_SPACES, TESTING_SPACES_IDS
@@ -110,7 +110,9 @@ def test_flatten(space):
assert single_dim == flatdim
else:
assert isinstance(flattened_sample, (tuple, dict, Graph))
assert isinstance(space, Sequence) or isinstance(
flattened_sample, (tuple, dict, Graph)
)
@pytest.mark.parametrize("space", TESTING_SPACES, ids=TESTING_SPACES_IDS)