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

Fix OneOf shared memory and add pytest.skip to tests (#999)

Browse Source
This commit is contained in:
Mark Towers
2024-04-06 13:20:10 +01:00
committed by GitHub
parent 9c812af180
commit 0fe94efa26
10 changed files with 50 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gymnasium/spaces/utils.py
View File

@@ -421,7 +421,7 @@ def _unflatten_sequence(space: Sequence, x: tuple[Any, ...]) -> tuple[Any, ...]
@unflatten.register(OneOf)
def _unflatten_oneof(space: OneOf, x: NDArray[Any]) -> tuple[int, Any]:
idx = int(x[0])
idx = np.int64(x[0])
sub_space = space.spaces[idx]
original_size = flatdim(sub_space)