API Reference of gxm#

Core API#

gxm.Dynamics(*args, **kwargs)

Base class for world models in gxm.

gxm.DynamicsState()

A placeholder class for dynamics/environment state.

gxm.Step(next_obs, action, info)

Output of a single model step: a pure dynamics transition.

gxm.Environment(*args, **kwargs)

Base class for RL environments in gxm.

gxm.EnvironmentState

alias of DynamicsState

gxm.Timestep(next_obs, action, info, reward, ...)

Extends Step with episodic metadata.

gxm.Transition(obs, action, reward, ...)

Class representing a single transition \((S_i, A_i, R_i, S_{i+1})\) in an environment.

gxm.Trajectory(obs, true_obs, action, ...)

Class representing a trajectory \((S_0, A_0, R_0, S_1, ..., S_n)\) in an environment.

Modules#

gxm.wrappers

Wrappers for gxm environments.

gxm.spaces

Spaces for defining action and observation spaces.