SYMBOL INDEX (218 symbols across 23 files)

FILE: baselines/baseline_fast_minimal.py
  function make_env (line 15) | def make_env(rank, seed=0):

FILE: baselines/global_map.py
  function local_to_global (line 17) | def local_to_global(r: int, c: int, map_n: int):

FILE: baselines/ray_exp/red_gym_env_ray.py
  class RedGymEnv (line 27) | class RedGymEnv(gym.Env):
    method __init__ (line 30) | def __init__(
    method reset (line 112) | def reset(self, *, seed=None, options=None):
    method init_knn (line 152) | def init_knn(self):
    method render (line 159) | def render(self, reduce_res=True, add_memory=True, update_mem=True):
    method step (line 180) | def step(self, action):
    method run_action_on_emulator (line 216) | def run_action_on_emulator(self, action):
    method add_video_frame (line 236) | def add_video_frame(self):
    method append_agent_stats (line 240) | def append_agent_stats(self):
    method update_frame_knn_index (line 254) | def update_frame_knn_index(self, frame_vec):
    method update_reward (line 274) | def update_reward(self):
    method group_rewards (line 292) | def group_rewards(self):
    method create_exploration_memory (line 299) | def create_exploration_memory(self):
    method create_recent_memory (line 328) | def create_recent_memory(self):
    method check_if_done (line 334) | def check_if_done(self):
    method save_and_print_info (line 344) | def save_and_print_info(self, done, obs_memory):
    method read_m (line 380) | def read_m(self, addr):
    method read_bit (line 383) | def read_bit(self, addr, bit: int) -> bool:
    method get_levels_sum (line 387) | def get_levels_sum(self):
    method get_levels_reward (line 391) | def get_levels_reward(self):
    method get_knn_reward (line 402) | def get_knn_reward(self):
    method get_badges (line 410) | def get_badges(self):
    method read_party (line 413) | def read_party(self):
    method update_heal_reward (line 416) | def update_heal_reward(self):
    method get_all_events_reward (line 428) | def get_all_events_reward(self):
    method get_game_state_reward (line 431) | def get_game_state_reward(self, print_stats=False):
    method save_screenshot (line 472) | def save_screenshot(self, name):
    method update_max_op_level (line 479) | def update_max_op_level(self):
    method update_max_event_rew (line 487) | def update_max_event_rew(self):
    method read_hp_fraction (line 492) | def read_hp_fraction(self):
    method read_hp (line 497) | def read_hp(self, start):
    method bit_count (line 501) | def bit_count(self, bits):
    method read_triple (line 504) | def read_triple(self, start_add):
    method read_bcd (line 507) | def read_bcd(self, num):
    method read_money (line 510) | def read_money(self):

FILE: baselines/red_gym_env.py
  class RedGymEnv (line 23) | class RedGymEnv(Env):
    method __init__ (line 26) | def __init__(
    method reset (line 118) | def reset(self, seed=None, options=None):
    method init_knn (line 163) | def init_knn(self):
    method init_map_mem (line 170) | def init_map_mem(self):
    method render (line 173) | def render(self, reduce_res=True, add_memory=True, update_mem=True):
    method step (line 194) | def step(self, action):
    method run_action_on_emulator (line 233) | def run_action_on_emulator(self, action):
    method add_video_frame (line 258) | def add_video_frame(self):
    method append_agent_stats (line 262) | def append_agent_stats(self, action):
    method update_frame_knn_index (line 285) | def update_frame_knn_index(self, frame_vec):
    method update_seen_coords (line 306) | def update_seen_coords(self):
    method update_reward (line 318) | def update_reward(self):
    method group_rewards (line 336) | def group_rewards(self):
    method create_exploration_memory (line 346) | def create_exploration_memory(self):
    method create_recent_memory (line 379) | def create_recent_memory(self):
    method check_if_done (line 385) | def check_if_done(self):
    method save_and_print_info (line 395) | def save_and_print_info(self, done, obs_memory):
    method read_m (line 431) | def read_m(self, addr):
    method read_bit (line 434) | def read_bit(self, addr, bit: int) -> bool:
    method get_levels_sum (line 438) | def get_levels_sum(self):
    method get_levels_reward (line 442) | def get_levels_reward(self):
    method get_knn_reward (line 453) | def get_knn_reward(self):
    method get_badges (line 462) | def get_badges(self):
    method read_party (line 465) | def read_party(self):
    method update_heal_reward (line 468) | def update_heal_reward(self):
    method get_all_events_reward (line 482) | def get_all_events_reward(self):
    method get_game_state_reward (line 500) | def get_game_state_reward(self, print_stats=False):
    method save_screenshot (line 541) | def save_screenshot(self, name):
    method update_max_op_level (line 548) | def update_max_op_level(self):
    method update_max_event_rew (line 556) | def update_max_event_rew(self):
    method read_hp_fraction (line 561) | def read_hp_fraction(self):
    method read_hp (line 567) | def read_hp(self, start):
    method bit_count (line 571) | def bit_count(self, bits):
    method read_triple (line 574) | def read_triple(self, start_add):
    method read_bcd (line 577) | def read_bcd(self, num):
    method read_money (line 580) | def read_money(self):
    method get_map_location (line 585) | def get_map_location(self, map_idx):

FILE: baselines/red_gym_env_minimal.py
  class PokeRedEnv (line 20) | class PokeRedEnv(Env):
    method __init__ (line 21) | def __init__(
    method reset (line 92) | def reset(self, seed=0, options={}):
    method init_map_mem (line 124) | def init_map_mem(self):
    method render (line 127) | def render(self, reduce_res=True):
    method _get_obs (line 136) | def _get_obs(self):
    method step (line 146) | def step(self, action):
    method run_action_on_emulator (line 197) | def run_action_on_emulator(self, action):
    method append_agent_stats (line 213) | def append_agent_stats(self, action):
    method get_game_coords (line 240) | def get_game_coords(self):
    method update_seen_coords (line 243) | def update_seen_coords(self):
    method get_global_coords (line 248) | def get_global_coords(self):
    method update_explore_map (line 255) | def update_explore_map(self):
    method get_explore_map (line 262) | def get_explore_map(self):
    method check_if_done (line 273) | def check_if_done(self):
    method read_m (line 278) | def read_m(self, addr):
    method read_bit (line 281) | def read_bit(self, addr, bit: int) -> bool:
    method read_event_bits (line 285) | def read_event_bits(self):
    method get_levels_sum (line 291) | def get_levels_sum(self):
    method get_badges (line 300) | def get_badges(self):
    method read_party (line 303) | def read_party(self):
    method get_all_events_reward (line 309) | def get_all_events_reward(self):
    method update_max_op_level (line 321) | def update_max_op_level(self):
    method update_heal_reward (line 333) | def update_heal_reward(self):
    method read_hp_fraction (line 343) | def read_hp_fraction(self):
    method read_hp (line 355) | def read_hp(self, start):
    method bit_count (line 359) | def bit_count(self, bits):
    method update_map_progress (line 362) | def update_map_progress(self):
    method get_map_progress (line 366) | def get_map_progress(self, map_idx):
    method get_map_location (line 372) | def get_map_location(self, map_idx):

FILE: baselines/render_all_needed_grids.py
  function make_env (line 12) | def make_env(rank, env_conf, seed=0):
  function run_save (line 27) | def run_save(save):

FILE: baselines/run_baseline_parallel.py
  function make_env (line 11) | def make_env(rank, env_conf, seed=0):

FILE: baselines/run_baseline_parallel_fast.py
  function make_env (line 12) | def make_env(rank, env_conf, seed=0):

FILE: baselines/run_pretrained_interactive.py
  function make_env (line 11) | def make_env(rank, env_conf, seed=0):

FILE: baselines/run_recorded_actions.py
  function run_recorded_actions_on_emulator_and_save_video (line 6) | def run_recorded_actions_on_emulator_and_save_video(sess_id, instance_id...

FILE: baselines/stream_agent_wrapper.py
  class StreamWrapper (line 10) | class StreamWrapper(gym.Wrapper):
    method __init__ (line 11) | def __init__(self, env, stream_metadata={}):
    method step (line 32) | def step(self, action):
    method broadcast_ws_message (line 58) | async def broadcast_ws_message(self, message):
    method establish_wc_connection (line 67) | async def establish_wc_connection(self):

FILE: baselines/tensorboard_callback.py
  function merge_dicts (line 10) | def merge_dicts(dicts):
  class TensorboardCallback (line 29) | class TensorboardCallback(BaseCallback):
    method __init__ (line 31) | def __init__(self, log_dir, verbose=0):
    method _on_training_start (line 36) | def _on_training_start(self):
    method _on_step (line 40) | def _on_step(self) -> bool:
    method _on_training_end (line 71) | def _on_training_end(self):

FILE: baselines/tile_vids_to_grid.py
  function run_ffmpeg_grid (line 9) | def run_ffmpeg_grid(out_path, files, screen_res_str, full_res_string, gx...
  function make_script (line 60) | def make_script(path):
  function make_outer_script (line 69) | def make_outer_script(out_file, paths):
  function write_file (line 74) | def write_file(out_file, script):

FILE: v2/baseline_fast_v2.py
  function make_env (line 13) | def make_env(rank, env_conf, seed=0):

FILE: v2/global_map.py
  function local_to_global (line 17) | def local_to_global(r: int, c: int, map_n: int):

FILE: v2/red_gym_env_v2.py
  class RedGymEnv (line 22) | class RedGymEnv(Env):
    method __init__ (line 23) | def __init__(self, config=None):
    method reset (line 123) | def reset(self, seed=None, options={}):
    method init_map_mem (line 167) | def init_map_mem(self):
    method render (line 170) | def render(self, reduce_res=True):
    method _get_obs (line 178) | def _get_obs(self):
    method step (line 201) | def step(self, action):
    method run_action_on_emulator (line 249) | def run_action_on_emulator(self, action):
    method append_agent_stats (line 262) | def append_agent_stats(self, action):
    method start_video (line 288) | def start_video(self):
    method add_video_frame (line 323) | def add_video_frame(self):
    method get_game_coords (line 334) | def get_game_coords(self):
    method update_seen_coords (line 337) | def update_seen_coords(self):
    method get_current_coord_count_reward (line 348) | def get_current_coord_count_reward(self):
    method get_global_coords (line 357) | def get_global_coords(self):
    method update_explore_map (line 361) | def update_explore_map(self):
    method get_explore_map (line 369) | def get_explore_map(self):
    method update_recent_screens (line 380) | def update_recent_screens(self, cur_screen):
    method update_recent_actions (line 384) | def update_recent_actions(self, action):
    method update_reward (line 388) | def update_reward(self):
    method group_rewards (line 399) | def group_rewards(self):
    method check_if_done (line 408) | def check_if_done(self):
    method save_and_print_info (line 413) | def save_and_print_info(self, done, obs):
    method read_m (line 459) | def read_m(self, addr):
    method read_bit (line 463) | def read_bit(self, addr, bit: int) -> bool:
    method read_event_bits (line 467) | def read_event_bits(self):
    method get_levels_sum (line 473) | def get_levels_sum(self):
    method get_levels_reward (line 482) | def get_levels_reward(self):
    method get_badges (line 493) | def get_badges(self):
    method read_party (line 496) | def read_party(self):
    method get_all_events_reward (line 502) | def get_all_events_reward(self):
    method get_game_state_reward (line 514) | def get_game_state_reward(self, print_stats=False):
    method update_max_op_level (line 530) | def update_max_op_level(self):
    method update_max_event_rew (line 542) | def update_max_event_rew(self):
    method update_heal_reward (line 547) | def update_heal_reward(self):
    method read_hp_fraction (line 557) | def read_hp_fraction(self):
    method read_hp (line 569) | def read_hp(self, start):
    method bit_count (line 573) | def bit_count(self, bits):
    method fourier_encode (line 576) | def fourier_encode(self, val):
    method update_map_progress (line 579) | def update_map_progress(self):
    method get_map_progress (line 583) | def get_map_progress(self, map_idx):

FILE: v2/run_pretrained_interactive.py
  function make_env (line 14) | def make_env(rank, env_conf, seed=0):
  function get_most_recent_zip_with_age (line 29) | def get_most_recent_zip_with_age(folder_path):

FILE: v2/stream_agent_wrapper.py
  class StreamWrapper (line 10) | class StreamWrapper(gym.Wrapper):
    method __init__ (line 11) | def __init__(self, env, stream_metadata={}):
    method step (line 32) | def step(self, action):
    method broadcast_ws_message (line 58) | async def broadcast_ws_message(self, message):
    method establish_wc_connection (line 67) | async def establish_wc_connection(self):

FILE: v2/tensorboard_callback.py
  function merge_dicts (line 10) | def merge_dicts(dicts):
  class TensorboardCallback (line 29) | class TensorboardCallback(BaseCallback):
    method __init__ (line 31) | def __init__(self, log_dir, verbose=0):
    method _on_training_start (line 36) | def _on_training_start(self):
    method _on_step (line 40) | def _on_step(self) -> bool:
    method _on_training_end (line 71) | def _on_training_end(self):

FILE: visualization/BetterMapVis_script_version.py
  function make_all_coords_arrays (line 15) | def make_all_coords_arrays(filtered_dfs):
  function load_tex (line 18) | def load_tex(name):
  function get_sprite_by_coords (line 22) | def get_sprite_by_coords(img, x, y):
  function game_coord_to_pixel_coord (line 29) | def game_coord_to_pixel_coord(
  function add_sprite (line 77) | def add_sprite(overlay_map, sprite, coord):
  function blend_overlay (line 91) | def blend_overlay(background, over):
  function split (line 97) | def split(img):
  function render_video (line 100) | def render_video(fname, all_coords, walks, bg, inter_steps=4, add_start=...
  function test_render (line 171) | def test_render(name, dat, walks, bg):

FILE: visualization/BetterMapVis_script_version_FLOW.py
  function make_all_coords_arrays (line 17) | def make_all_coords_arrays(filtered_dfs):
  function get_sprite_by_coords (line 20) | def get_sprite_by_coords(img, x, y):
  function game_coord_to_global_coord (line 27) | def game_coord_to_global_coord(
  function add_sprite (line 75) | def add_sprite(overlay_map, sprite, coord):
  function blend_overlay (line 89) | def blend_overlay(background, over):
  function split (line 95) | def split(img):
  function compute_flow (line 98) | def compute_flow(all_coords, inter_steps=1, add_start=True):
  function render_arrows (line 171) | def render_arrows(fname, all_flows, arrow_sprite):
  function compute_flow_wrap (line 228) | def compute_flow_wrap(dat):

FILE: visualization/BetterMapVis_script_version_FLOW_edge.py
  function make_all_coords_arrays (line 17) | def make_all_coords_arrays(filtered_dfs):
  function get_sprite_by_coords (line 20) | def get_sprite_by_coords(img, x, y):
  function game_coord_to_global_coord (line 27) | def game_coord_to_global_coord(
  function add_sprite (line 75) | def add_sprite(overlay_map, sprite, coord):
  function blend_overlay (line 89) | def blend_overlay(background, over):
  function split (line 95) | def split(img):
  function compute_flow (line 98) | def compute_flow(all_coords, inter_steps=1, add_start=True):
  function render_arrows (line 171) | def render_arrows(fname, all_flows, arrow_sprite):
  function compute_flow_wrap (line 258) | def compute_flow_wrap(dat):

FILE: visualization/BetterMapVis_script_version_PROG_COLOR.py
  function make_all_coords_arrays (line 19) | def make_all_coords_arrays(filtered_dfs):
  function load_tex (line 22) | def load_tex(name):
  function get_sprite_by_coords (line 26) | def get_sprite_by_coords(img, x, y):
  function game_coord_to_pixel_coord (line 33) | def game_coord_to_pixel_coord(
  function add_sprite (line 81) | def add_sprite(overlay_map, sprite, coord):
  function blend_overlay (line 95) | def blend_overlay(background, over):
  function split (line 101) | def split(img):
  function render_video (line 104) | def render_video(fname, all_coords, walks, bg, inter_steps=4, add_start=...
  function test_render (line 177) | def test_render(name, dat, walks, bg):