unilab.base.observations.get_obs_dims

unilab.base.observations.get_obs_dims(obs_groups_spec)[source]

Extract (actor_obs_dim, critic_obs_dim) from obs_groups_spec.

When no separate critic group exists, critic_obs_dim == actor_obs_dim.

Parameters:

obs_groups_spec (dict[str, int])

Return type:

tuple[int, int]