mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-05 15:31:44 +00:00
6332d4f113
`contains` really should not exist when it does exactly what the builtin magic method `__contains__` was meant for, but that would break backward compatibility.
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import gym
|
|
|
|
class Tuple(gym.Space):
|
|
"""
|
|
A tuple (i.e., product) of simpler spaces
|
|
|
|
Example usage:
|
|
self.observation_space = spaces.Tuple((spaces.Discrete(2), spaces.Discrete(3)))
|
|
"""
|
|
def __init__(self, spaces):
|
|
self.spaces = spaces
|
|
gym.Space.__init__(self, None, None)
|
|
|
|
def sample(self):
|
|
return tuple([space.sample() for space in self.spaces])
|
|
|
|
def contains(self, x):
|
|
if isinstance(x, list):
|
|
x = tuple(x) # Promote list to tuple for contains check
|
|
return isinstance(x, tuple) and len(x) == len(self.spaces) and all(
|
|
space.contains(part) for (space,part) in zip(self.spaces,x))
|
|
|
|
def __repr__(self):
|
|
return "Tuple(" + ", ". join([str(s) for s in self.spaces]) + ")"
|
|
|
|
def to_jsonable(self, sample_n):
|
|
# serialize as list-repr of tuple of vectors
|
|
return [space.to_jsonable([sample[i] for sample in sample_n]) \
|
|
for i, space in enumerate(self.spaces)]
|
|
|
|
def from_jsonable(self, sample_n):
|
|
return [sample for sample in zip(*[space.from_jsonable(sample_n[i]) for i, space in enumerate(self.spaces)])]
|