def gym_space_from_arrays(arrays):
""" Define environment observation space using an example observation """
if isinstance(arrays, np.ndarray):
ret = Box(-np.inf, np.inf, arrays.shape, np.float32)
ret.flatten_dim = np.prod(ret.shape)
elif isinstance(arrays, (tuple, list)):
ret = Tuple([gym_space_from_arrays(arr) for arr in arrays])
elif isinstance(arrays, dict):
ret = Dict(dict([(k, gym_space_from_arrays(v)) for k, v in arrays.items()]))
else:
raise TypeError(f"Array is of unsupported type: {type(arrays)}")
return ret
def gym_space_from_arrays(arrays):
if isinstance(arrays, np.ndarray):
ret = Box(-np.inf, np.inf, arrays.shape, np.float32)
ret.flatten_dim = np.prod(ret.shape)
elif isinstance(arrays, (tuple, list)):
ret = Tuple([gym_space_from_arrays(arr) for arr in arrays])
elif isinstance(arrays, dict):
ret = Dict(
dict([(k, gym_space_from_arrays(v)) for k, v in arrays.items()]))
else:
raise TypeError("Array is of unsupported type.")
return ret
