Skip to content

ochre_gym.spaces

OchreEquipmentSubSpace

ochre_gym.spaces.OchreEquipmentSubSpace

Class for keeping track of attributes for Ochre Equipment actions/obs subspaces.

__init__(name: str, shape: Tuple, type: SubSpaceType, upper_bound: np.array = np.array([1.0]), lower_bound: np.array = np.array([0.0]), n: int = 2)

Parameters:

Name Type Description Default
name str

Name of the subspace

required
shape Tuple

Shape of the subspace

required
type SubSpaceType

Type of the subspace

required
upper_bound np.array

Upper bound of the subspace

np.array([1.0])
lower_bound np.array

Lower bound of the subspace

np.array([0.0])
n int

Number of discrete values in the subspace

2
register()

Returns a dictionary of the attributes of the subspace.

SubSpaceType

ochre_gym.spaces.SubSpaceType

Enum for the action type of the subspace.

register_equipment_subspace

ochre_gym.spaces.register_equipment_subspace(list_of_dicts: List)

Converts a list of dictionaries to a dictionary of dictionaries.

Parameters:

Name Type Description Default
list_of_dicts list

A list of dictionaries

required

Returns:

Name Type Description
subspace Dict

A dictionary of dictionaries