StepStateCriticComposer

class maze.perception.models.critics.step_state_critic_composer.StepStateCriticComposer(observation_spaces_dict: Dict[str | int, gymnasium.spaces.Dict], agent_counts_dict: Dict[str | int, int], networks: List[None | Mapping[str, Any] | Any] | Mapping[str | Type, None | Mapping[str, Any] | Any])

Each sub-step or actor gets its individual critic.

Instantiates a TorchStepStateCritic.

Parameters:

• observation_spaces_dict – Dict of sub-step id to observation space.

• networks – Critics networks as defined in the config (either list or dictionary of object params and type).

property critic: TorchStepStateCritic

(overrides BaseStateCriticComposer)

implementation of BaseStateCriticComposer