WorldModelPolicy

class WorldModelPolicy(solver: Solver, config: PlanConfig, process: dict[str, Transformable] | None = None, transform: dict[str, callable] | None = None, **kwargs)

Bases: BasePolicy

World Model Policy using a planning solver.

property flatten_receding_horizon

get_action(info_dict, **kwargs)

Get action from the policy given the observation.

set_env(env)