mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-01 06:07:08 +00:00
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from gym import Space
|
|
import numpy as np
|
|
|
|
class Box(Space):
|
|
"""
|
|
A box in R^n.
|
|
I.e., each coordinate is bounded.
|
|
"""
|
|
def __init__(self, low, high, shape=None):
|
|
"""
|
|
Two kinds of valid input:
|
|
Box(-1.0, 1.0, (3,4)) # low and high are scalars, and shape is provided
|
|
Box(np.array([-1.0,-2.0]), np.array([2.0,4.0])) # low and high are arrays of the same shape
|
|
"""
|
|
if shape is None:
|
|
assert low.shape == high.shape
|
|
self.low = low
|
|
self.high = high
|
|
else:
|
|
assert np.isscalar(low) and np.isscalar(high)
|
|
self.low = low + np.zeros(shape)
|
|
self.high = high + np.zeros(shape)
|
|
def sample(self):
|
|
return np.random.uniform(low=self.low, high=self.high, size=self.low.shape)
|
|
def contains(self, x):
|
|
return x.shape == self.shape and (x >= self.low).all() and (x <= self.high).all()
|
|
|
|
def to_jsonable(self, sample_n):
|
|
return np.array(sample_n).tolist()
|
|
def from_jsonable(self, sample_n):
|
|
return [np.asarray(sample) for sample in sample_n]
|
|
|
|
@property
|
|
def shape(self):
|
|
return self.low.shape
|
|
def __repr__(self):
|
|
return "Box" + str(self.shape)
|
|
def __eq__(self, other):
|
|
return np.allclose(self.low, other.low) and np.allclose(self.high, other.high)
|