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