mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-01 22:11:25 +00:00
* add dtype to Box * remove board_game, debugging, safety, parameter_tuning environments * massive set of breaking changes - remove python logging module - _step, _reset, _seed, _close => non underscored method - remove benchmark and scoring folder * Improve render("human"), now resizable, closable window. * get rid of default step and reset in wrappers, so it doesn’t silently fail for people with underscore methods * CubeCrash unit test environment * followup fixes * MemorizeDigits unit test envrionment * refactored spaces a bit fixed indentation disabled test_env_semantics * fix unit tests * fixes * CubeCrash, MemorizeDigits tested * gym backwards compatibility patch * gym backwards compatibility, followup fixes * changelist, add spaces to main namespaces * undo_logger_setup for backwards compat * remove configuration.py
15 lines
523 B
Python
15 lines
523 B
Python
from gym import spaces, Space
|
|
import numpy as np
|
|
|
|
class MultiBinary(Space):
|
|
def __init__(self, n):
|
|
self.n = n
|
|
Space.__init__(self, (self.n,), np.int8)
|
|
def sample(self):
|
|
return spaces.np_random.randint(low=0, high=2, size=self.n).astype(self.dtype)
|
|
def contains(self, x):
|
|
return ((x==0) | (x==1)).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] |