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

Update vector space utility functions for all spaces (#223)

Browse Source
This commit is contained in:
Mark Towers
2023-02-22 15:05:58 +00:00
committed by GitHub
parent 761bb2e033
commit a7d9146b1d
8 changed files with 293 additions and 408 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gymnasium/spaces/utils.py
View File

@@ -243,7 +243,7 @@ def _flatten_sequence(
space: Sequence, x: tuple[Any, ...] | Any
) -> tuple[Any, ...] | Any:
if space.stack:
samples_iters = gym.vector.utils.iterate(space.batched_feature_space, x)
samples_iters = gym.vector.utils.iterate(space.stacked_feature_space, x)
flattened_samples = [
flatten(space.feature_space, sample) for sample in samples_iters
]