SYMBOL INDEX (103 symbols across 9 files)

FILE: gui/recording.py
  class SaveNpzWrapper (line 11) | class SaveNpzWrapper(gym.Wrapper):
    method __init__ (line 13) | def __init__(self, env, log_dir, video_fps=30, video_size=256, video_f...
    method step (line 23) | def step(self, action):
    method _save_npz (line 36) | def _save_npz(self, data, path):
    method _save_video (line 41) | def _save_video(self, data, path):
  class CollectWrapper (line 51) | class CollectWrapper(gym.Wrapper):
    method __init__ (line 54) | def __init__(self, env):
    method step (line 60) | def step(self, action):
    method reset (line 69) | def reset(self):
  class ActionRewardResetWrapper (line 76) | class ActionRewardResetWrapper(gym.Wrapper):
    method __init__ (line 79) | def __init__(self, env, no_terminal=False):
    method step (line 86) | def step(self, action):
    method reset (line 100) | def reset(self):

FILE: gui/run_gui.py
  function get_keymap (line 28) | def get_keymap(env):
  function main (line 39) | def main():
  function obs_to_text (line 199) | def obs_to_text(obs, env, steps, return_):
  function keymap_to_text (line 209) | def keymap_to_text(keymap, verbose=False):

FILE: memory_maze/__init__.py
  function _make_gym_env (line 22) | def _make_gym_env(dm_task: Callable[[], dm_env.Environment], **kwargs):

FILE: memory_maze/gym_wrappers.py
  class GymWrapper (line 10) | class GymWrapper(gym.Env):
    method __init__ (line 12) | def __init__(self, env: dm_env.Environment):
    method reset (line 17) | def reset(self) -> Any:
    method step (line 21) | def step(self, action) -> Tuple[Any, float, bool, dict]:
  function _convert_to_space (line 33) | def _convert_to_space(spec: Any) -> gym.Space:

FILE: memory_maze/helpers.py
  function sample_spec (line 4) | def sample_spec(space: BoundedArray) -> np.ndarray:

FILE: memory_maze/maze.py
  class RollingBallWithFriction (line 29) | class RollingBallWithFriction(jumping_ball.RollingBallWithHead):
    method _build (line 31) | def _build(self, roll_damping=5.0, steer_damping=20.0, **kwargs):
  class MemoryMazeTask (line 39) | class MemoryMazeTask(random_goal_maze.NullGoalMaze):
    method __init__ (line 42) | def __init__(self,
    method task_observables (line 116) | def task_observables(self):
    method name (line 120) | def name(self):
    method initialize_episode_mjcf (line 123) | def initialize_episode_mjcf(self, rng: RandomState):
    method initialize_episode (line 137) | def initialize_episode(self, physics, rng: RandomState):
    method after_step (line 142) | def after_step(self, physics, rng: RandomState):
    method should_terminate_episode (line 153) | def should_terminate_episode(self, physics):
    method get_reward (line 156) | def get_reward(self, physics):
    method _create_targets (line 161) | def _create_targets(self, clear_existing=False, randomize_colors=False...
    method _place_targets (line 184) | def _place_targets(self, rng: RandomState) -> bool:
    method _pick_new_target (line 194) | def _pick_new_target(self, rng: RandomState):
  class FixedWallTexture (line 203) | class FixedWallTexture(labmaze_textures.WallTextures):
    method _build (line 206) | def _build(self, style, texture_name):
  class FixedFloorTexture (line 218) | class FixedFloorTexture(labmaze_textures.FloorTextures):
    method _build (line 221) | def _build(self, style, texture_names):
  class MazeWithTargetsArena (line 236) | class MazeWithTargetsArena(mazes.MazeWithTargets):
    method _build (line 239) | def _build(self,
    method regenerate (line 278) | def regenerate(self, random_state):
    method _make_floor_variations (line 312) | def _make_floor_variations(self, build_tile_geoms_fn=None):
  class TextMazeVaryingWalls (line 367) | class TextMazeVaryingWalls(labmaze.RandomMaze):
    method regenerate (line 370) | def regenerate(self):
    method _block_variations (line 374) | def _block_variations(self):
    method _change_block_char (line 389) | def _change_block_char(self, i1, i2, j1, j2, char):

FILE: memory_maze/oracle.py
  class PathToTargetWrapper (line 8) | class PathToTargetWrapper(ObservationWrapper):
    method observation_spec (line 11) | def observation_spec(self):
    method observation (line 19) | def observation(self, obs):
  class DrawMinimapWrapper (line 32) | class DrawMinimapWrapper(ObservationWrapper):
    method observation_spec (line 35) | def observation_spec(self):
    method observation (line 43) | def observation(self, obs):
  function breadth_first_search (line 78) | def breadth_first_search(maze: np.ndarray, start: Tuple[int, int], finis...

FILE: memory_maze/tasks.py
  function memory_maze_9x9 (line 14) | def memory_maze_9x9(**kwargs):
  function memory_maze_11x11 (line 28) | def memory_maze_11x11(**kwargs):
  function memory_maze_13x13 (line 32) | def memory_maze_13x13(**kwargs):
  function memory_maze_15x15 (line 36) | def memory_maze_15x15(**kwargs):
  function _memory_maze (line 50) | def _memory_maze(

FILE: memory_maze/wrappers.py
  class Wrapper (line 10) | class Wrapper(dm_env.Environment):
    method __init__ (line 13) | def __init__(self, env: dm_env.Environment):
    method __getattr__ (line 16) | def __getattr__(self, name):
    method step (line 21) | def step(self, action) -> dm_env.TimeStep:
    method reset (line 24) | def reset(self) -> dm_env.TimeStep:
    method action_spec (line 27) | def action_spec(self) -> Any:
    method discount_spec (line 30) | def discount_spec(self) -> Any:
    method observation_spec (line 33) | def observation_spec(self) -> Any:
    method reward_spec (line 36) | def reward_spec(self) -> Any:
    method close (line 39) | def close(self):
  class ObservationWrapper (line 43) | class ObservationWrapper(Wrapper):
    method observation_spec (line 46) | def observation_spec(self):
    method observation (line 49) | def observation(self, obs: Any) -> Any:
    method step (line 52) | def step(self, action) -> dm_env.TimeStep:
    method reset (line 56) | def reset(self) -> dm_env.TimeStep:
  class RemapObservationWrapper (line 61) | class RemapObservationWrapper(ObservationWrapper):
    method __init__ (line 64) | def __init__(self, env: dm_env.Environment, mapping: Dict[str, str]):
    method observation_spec (line 68) | def observation_spec(self):
    method observation (line 73) | def observation(self, obs):
  class TargetsPositionWrapper (line 78) | class TargetsPositionWrapper(ObservationWrapper):
    method __init__ (line 83) | def __init__(self, env: dm_env.Environment, maze_xy_scale, maze_width,...
    method observation_spec (line 101) | def observation_spec(self):
    method observation (line 112) | def observation(self, obs):
  class AgentPositionWrapper (line 129) | class AgentPositionWrapper(ObservationWrapper):
    method __init__ (line 132) | def __init__(self, env: dm_env.Environment, maze_xy_scale, maze_width,...
    method observation_spec (line 137) | def observation_spec(self):
    method observation (line 147) | def observation(self, obs):
  class MazeLayoutWrapper (line 159) | class MazeLayoutWrapper(ObservationWrapper):
    method observation_spec (line 162) | def observation_spec(self):
    method observation (line 171) | def observation(self, obs):
  class ImageOnlyObservationWrapper (line 181) | class ImageOnlyObservationWrapper(ObservationWrapper):
    method __init__ (line 184) | def __init__(self, env: dm_env.Environment, key: str = 'image'):
    method observation_spec (line 188) | def observation_spec(self):
    method observation (line 193) | def observation(self, obs):
  class DiscreteActionSetWrapper (line 198) | class DiscreteActionSetWrapper(Wrapper):
    method __init__ (line 201) | def __init__(self, env: dm_env.Environment, action_set: List[np.ndarra...
    method action_spec (line 205) | def action_spec(self):
    method step (line 208) | def step(self, action) -> dm_env.TimeStep:
  class TargetColorAsBorderWrapper (line 212) | class TargetColorAsBorderWrapper(ObservationWrapper):
    method observation_spec (line 215) | def observation_spec(self):
    method observation (line 221) | def observation(self, obs):