Index

_

__init__() (Box method), [1] (CubeEnv method) (Dict method), [1] (Discrete method), [1] (DrawOptions method) (MultiDiscrete method), [1] (RandomSolver method), [1], [2] (SceneEnv method) (StepsDataset method), [1] (World method), [1], [2]

A

Action (class in stable_worldmodel.envs.voidrun) action_block (PlanConfig attribute), [1], [2] action_dim (CEMSolver property), [1], [2] (GDSolver property), [1], [2] (MPPISolver property), [1], [2] (NevergradSolver property), [1], [2] (RandomSolver property), [1], [2] (Solver property), [1], [2] action_space (World property), [1], [2] (WorldInfo attribute) add_box() (PushT method) add_circle() (PushT method) (TwoRoomEnv method) add_I() (PushT method)

add_L() (PushT method) add_object_info() (CubeEnv method) (SceneEnv method) add_objects() (CubeEnv method) (SceneEnv method) add_plus() (PushT method) add_shape() (PushT method) add_small_tee() (PushT method) add_square() (PushT method) add_tee() (PushT method) add_Z() (PushT method) AddPixelsWrapper (class in stable_worldmodel.wrappers), [1] all_reduce() (in module stable_worldmodel.wm.dinowm) Attention (class in stable_worldmodel.wm.dinowm) AutoCostModel() (in module stable_worldmodel.policy)

B

BasePolicy (class in stable_worldmodel.policy), [1]

Box (class in stable_worldmodel.spaces), [1]

C

CausalPredictor (class in stable_worldmodel.wm.dinowm) CEMSolver (class in stable_worldmodel.solver), [1] (class in stable_worldmodel.solver.cem) check() (Box method), [1] (Dict method), [1] (Discrete method), [1] (MultiDiscrete method), [1] check_collide() (TwoRoomEnv method) check_location() (VoidRunEnv method) check_one_door_fit() (TwoRoomEnv method) check_other_room() (TwoRoomEnv method) check_termination() (VoidRunEnv method) close() (ImagePositioning method) (PushT method) (SimplePointMazeEnv method) (TwoRoomEnv method) (VoidRunEnv method) (World method), [1], [2] common() (in module stable_worldmodel.cli)

compute_observation() (CubeEnv method) (SceneEnv method) compute_oracle_observation() (CubeEnv method) (SceneEnv method) compute_reward() (CubeEnv method) (SceneEnv method) compute_trajectory_weights() (MPPISolver method), [1], [2] config (WorldInfo attribute) configure() (CEMSolver method), [1], [2] (GDSolver method), [1], [2] (MPPISolver method), [1], [2] (NevergradSolver method), [1], [2] (RandomSolver method), [1], [2] (Solver method), [1], [2] contains() (Box method), [1] (Dict method), [1] (Discrete method), [1] (MultiDiscrete method), [1] Costable (class in stable_worldmodel.solver.solver), [1] CubeEnv (class in stable_worldmodel.envs.ogbench_cube)

D

dataset_info() (in module stable_worldmodel.data) decode() (DINOWM method), [1] (VQVAE method) decode_code() (VQVAE method) Decoder (class in stable_worldmodel.wm.dinowm) delete() (in module stable_worldmodel.cli) delete_dataset() (in module stable_worldmodel.data) delete_model() (in module stable_worldmodel.data) Dict (class in stable_worldmodel.spaces), [1] DINOWM (class in stable_worldmodel.wm) (class in stable_worldmodel.wm.dinowm) Discrete (class in stable_worldmodel.spaces), [1]

display_dataset_info() (in module stable_worldmodel.cli) display_world_info() (in module stable_worldmodel.cli) DOWN (Action attribute) draw_circle() (DrawOptions method) draw_dot() (DrawOptions method) draw_fat_segment() (DrawOptions method) draw_polygon() (DrawOptions method) draw_segment() (DrawOptions method) DrawOptions (class in stable_worldmodel.envs.utils) dtype (SpaceInfo attribute) DummyWorldModel (class in stable_worldmodel.wm) (class in stable_worldmodel.wm.dummy)

E

embed_code() (Quantize method) Embedder (class in stable_worldmodel.wm.dinowm) encode() (DINOWM method), [1] (DummyWorldModel method), [1] Encoder (class in stable_worldmodel.wm.dinowm) EnsureGoalInfoWrapper (class in stable_worldmodel.wrappers), [1]

EnsureImageShape (class in stable_worldmodel.wrappers), [1] EnsureInfoKeysWrapper (class in stable_worldmodel.wrappers), [1] envs (VariationWrapper property), [1] eval_state() (PushT method) evaluate() (World method), [1], [2] EverythingToInfoWrapper (class in stable_worldmodel.wrappers), [1] ExpertPolicy (class in stable_worldmodel.policy), [1]

F

FeedForward (class in stable_worldmodel.wm.dinowm) fix_action_sample() (PushT method) flatten_dict() (in module stable_worldmodel.utils) flatten_receding_horizon (WorldModelPolicy property), [1] forward() (Attention method) (CausalPredictor method) (Decoder method) (Embedder method) (Encoder method) (FeedForward method) (Quantize method) (ResBlock method) (Transformer method) (VQVAE method)

from_pygame() (in module stable_worldmodel.envs.utils)

G

GDSolver (class in stable_worldmodel.solver), [1] (class in stable_worldmodel.solver.gd) generate_board() (VoidRunEnv method) generate_goal() (VoidRunEnv method) generate_mask_matrix() (Attention method) get_action() (BasePolicy method), [1] (ExpertPolicy method), [1] (RandomPolicy method), [1] (WorldModelPolicy method), [1] get_cache_dir() (in module stable_worldmodel.data)

get_cost() (Costable method), [1] (DINOWM method), [1] (DummyWorldModel method), [1] get_episode_slice() (StepsDataset method), [1] get_in() (in module stable_worldmodel.utils) get_mouse_pos() (in module stable_worldmodel.envs.utils) get_reset_info() (CubeEnv method) (SceneEnv method) get_step_info() (CubeEnv method) (SceneEnv method) get_world_size() (in module stable_worldmodel.wm.dinowm)

H

has_variation (VariationInfo attribute) high (SpaceInfo attribute) history_len (PlanConfig attribute), [1], [2] horizon (CEMSolver property), [1], [2] (GDSolver property), [1], [2] (MPPISolver property), [1], [2] (NevergradSolver property), [1], [2] (PlanConfig attribute), [1], [2] (RandomSolver property), [1], [2] (Solver property), [1], [2]

I

ImagePositioning (class in stable_worldmodel.envs.image_positioning) infer_img_path_columns() (StepsDataset method), [1] init_action() (GDSolver method), [1], [2] init_action_distrib() (CEMSolver method), [1], [2] (MPPISolver method), [1], [2] init_value (Box property), [1] (Dict property), [1] (Discrete property), [1] (MultiDiscrete property), [1]

initialize_episode() (CubeEnv method) (SceneEnv method) inverse_transform() (Transformable method) is_image() (in module stable_worldmodel.data)

L

LEFT (Action attribute) light_color() (in module stable_worldmodel.envs.utils) list_cmd() (in module stable_worldmodel.cli)

list_datasets() (in module stable_worldmodel.data) list_models() (in module stable_worldmodel.data) list_worlds() (in module stable_worldmodel.data) low (SpaceInfo attribute)

M

MegaWrapper (class in stable_worldmodel.wrappers), [1] metadata (ImagePositioning attribute) (PushT attribute) (SimplePointMazeEnv attribute) (TwoRoomEnv attribute) (VoidRunEnv attribute) modify_mjcf_model() (CubeEnv method) (SceneEnv method) module stable_worldmodel stable_worldmodel.cli stable_worldmodel.data stable_worldmodel.envs stable_worldmodel.envs.image_positioning stable_worldmodel.envs.ogbench_cube stable_worldmodel.envs.ogbench_scene stable_worldmodel.envs.pusht stable_worldmodel.envs.simple_point_maze stable_worldmodel.envs.two_room stable_worldmodel.envs.utils stable_worldmodel.envs.voidrun stable_worldmodel.policy stable_worldmodel.solver stable_worldmodel.solver.cem stable_worldmodel.solver.gd stable_worldmodel.solver.mppi stable_worldmodel.solver.nevergrad stable_worldmodel.solver.random stable_worldmodel.solver.solver, [1] stable_worldmodel.spaces stable_worldmodel.utils stable_worldmodel.wm stable_worldmodel.wm.dinowm stable_worldmodel.wm.dreamer stable_worldmodel.wm.dummy stable_worldmodel.wm.frame stable_worldmodel.wm.tdmpc stable_worldmodel.world stable_worldmodel.wrappers

MPPISolver (class in stable_worldmodel.solver), [1] (class in stable_worldmodel.solver.mppi) MultiDiscrete (class in stable_worldmodel.spaces), [1]

N

n (SpaceInfo attribute) n_envs (CEMSolver property), [1], [2] (GDSolver property), [1], [2] (MPPISolver property), [1], [2] (NevergradSolver property), [1], [2] (RandomSolver property), [1], [2] (Solver property), [1], [2]

name (WorldInfo attribute) names (VariationInfo attribute) names() (Dict method), [1] NevergradSolver (class in stable_worldmodel.solver), [1] (class in stable_worldmodel.solver.nevergrad) num_envs (World property), [1], [2]

O

observation_space (World property), [1], [2] (WorldInfo attribute)

P

perturb_camera_angle() (in module stable_worldmodel.envs.utils) plan_len (PlanConfig property), [1], [2] PlanConfig (class in stable_worldmodel) (class in stable_worldmodel.policy), [1] post_compilation_objects() (CubeEnv method) (SceneEnv method) post_step() (CubeEnv method) (SceneEnv method)

pre_step() (SceneEnv method) predict() (DINOWM method), [1] (DummyWorldModel method), [1] pretraining() (in module stable_worldmodel) (in module stable_worldmodel.utils) PushT (class in stable_worldmodel.envs.pusht) pymunk_to_shapely() (in module stable_worldmodel.envs.utils)

Q

Quantize (class in stable_worldmodel.wm.dinowm)

R

RandomPolicy (class in stable_worldmodel.policy), [1] RandomSolver (class in stable_worldmodel.solver), [1] (class in stable_worldmodel.solver.random) receding_horizon (PlanConfig attribute), [1], [2] record_dataset() (World method), [1], [2] record_video() (World method), [1], [2] record_video_from_dataset() (World method), [1], [2] register() (in module stable_worldmodel.envs) render() (CubeEnv method) (ImagePositioning method) (PushT method) (SceneEnv method) (SimplePointMazeEnv method) (TwoRoomEnv method) (VoidRunEnv method) render_board() (VoidRunEnv method) render_multiview() (CubeEnv method) replace_action_in_embedding() (DINOWM method), [1] ResBlock (class in stable_worldmodel.wm.dinowm) reset() (AddPixelsWrapper method), [1] (Box method), [1] (CubeEnv method) (Dict method), [1] (Discrete method), [1] (EnsureGoalInfoWrapper method), [1] (EnsureImageShape method), [1] (EnsureInfoKeysWrapper method), [1] (EverythingToInfoWrapper method), [1] (ImagePositioning method) (MegaWrapper method), [1] (MultiDiscrete method), [1] (PushT method) (ResizeGoalWrapper method), [1] (SceneEnv method) (SimplePointMazeEnv method) (TwoRoomEnv method) (VoidRunEnv method) (World method), [1], [2]

ResizeGoalWrapper (class in stable_worldmodel.wrappers), [1] reward_range (PushT attribute) RGBBox (class in stable_worldmodel.spaces), [1] RIGHT (Action attribute) rollout() (DINOWM method), [1]

S

sample() (Box method), [1] (Dict method), [1] (Discrete method), [1] (MultiDiscrete method), [1] sampling_order (Dict property), [1] SceneEnv (class in stable_worldmodel.envs.ogbench_scene) seed() (TwoRoomEnv method) set_env() (BasePolicy method), [1] (WorldModelPolicy method), [1] set_new_target() (CubeEnv method) (SceneEnv method) set_policy() (World method), [1], [2] set_seed() (GDSolver method), [1], [2] (RandomPolicy method), [1] set_state() (SceneEnv method) (VoidRunEnv method) set_tasks() (CubeEnv method) (SceneEnv method) shape (SpaceInfo attribute) show() (in module stable_worldmodel.cli) SimplePointMazeEnv (class in stable_worldmodel.envs.simple_point_maze) single_action_space (World property), [1], [2] single_observation_space (World property), [1], [2] single_variation_space (World property), [1], [2] solve() (CEMSolver method), [1], [2] (GDSolver method), [1], [2] (MPPISolver method), [1], [2] (NevergradSolver method), [1], [2] (RandomSolver method), [1], [2] (Solver method), [1], [2] Solver (class in stable_worldmodel.solver) (class in stable_worldmodel.solver.solver), [1] SpaceInfo (class in stable_worldmodel.data) split_embedding() (DINOWM method), [1] stable_worldmodel module stable_worldmodel.cli module stable_worldmodel.data module stable_worldmodel.envs module stable_worldmodel.envs.image_positioning module stable_worldmodel.envs.ogbench_cube module stable_worldmodel.envs.ogbench_scene module stable_worldmodel.envs.pusht module stable_worldmodel.envs.simple_point_maze module stable_worldmodel.envs.two_room module

stable_worldmodel.envs.utils module stable_worldmodel.envs.voidrun module stable_worldmodel.policy module stable_worldmodel.solver module stable_worldmodel.solver.cem module stable_worldmodel.solver.gd module stable_worldmodel.solver.mppi module stable_worldmodel.solver.nevergrad module stable_worldmodel.solver.random module stable_worldmodel.solver.solver module, [1] stable_worldmodel.spaces module stable_worldmodel.utils module stable_worldmodel.wm module stable_worldmodel.wm.dinowm module stable_worldmodel.wm.dreamer module stable_worldmodel.wm.dummy module stable_worldmodel.wm.frame module stable_worldmodel.wm.tdmpc module stable_worldmodel.world module stable_worldmodel.wrappers module step() (AddPixelsWrapper method), [1] (EnsureGoalInfoWrapper method), [1] (EnsureImageShape method), [1] (EnsureInfoKeysWrapper method), [1] (EverythingToInfoWrapper method), [1] (ImagePositioning method) (MegaWrapper method), [1] (PushT method) (ResizeGoalWrapper method), [1] (SimplePointMazeEnv method) (TwoRoomEnv method) (VoidRunEnv method) (World method), [1], [2] StepsDataset (class in stable_worldmodel.data), [1]

T

to_pygame() (in module stable_worldmodel.envs.utils) transform() (in module stable_worldmodel.wm.dummy) (Transformable method) Transformable (class in stable_worldmodel.policy)

Transformer (class in stable_worldmodel.wm.dinowm) TwoRoomEnv (class in stable_worldmodel.envs.two_room) type (SpaceInfo attribute) (VariationInfo attribute)

U

UP (Action attribute)

update() (Dict method), [1]

V

value (Box property), [1] (Dict property), [1] (Discrete property), [1] (MultiDiscrete property), [1] variation (WorldInfo attribute)

variation_space (World property), [1], [2] VariationInfo (class in stable_worldmodel.data) VariationWrapper (class in stable_worldmodel.wrappers), [1] VoidRunEnv (class in stable_worldmodel.envs.voidrun) VQVAE (class in stable_worldmodel.wm.dinowm)

W

warm_start (PlanConfig attribute), [1], [2] World (class in stable_worldmodel) (class in stable_worldmodel.world), [1]

world_info() (in module stable_worldmodel.data) WorldInfo (class in stable_worldmodel.data) WorldModelPolicy (class in stable_worldmodel.policy), [1]