SYMBOL INDEX (972 symbols across 60 files)

FILE: timechamber/__init__.py
  function make (line 14) | def make(

FILE: timechamber/ase/ase_agent.py
  class ASEAgent (line 40) | class ASEAgent(amp_agent.AMPAgent):
    method __init__ (line 41) | def __init__(self, base_name, config):
    method init_tensors (line 45) | def init_tensors(self):
    method play_steps (line 64) | def play_steps(self):
    method get_action_values (line 145) | def get_action_values(self, obs_dict, ase_latents, rand_action_probs):
    method prepare_dataset (line 178) | def prepare_dataset(self, batch_dict):
    method calc_gradients (line 186) | def calc_gradients(self, input_dict):
    method env_reset (line 337) | def env_reset(self, env_ids=None):
    method _reset_latent_step_count (line 350) | def _reset_latent_step_count(self, env_ids):
    method _load_config_params (line 355) | def _load_config_params(self, config):
    method _build_net_config (line 374) | def _build_net_config(self):
    method _reset_latents (line 379) | def _reset_latents(self, env_ids):
    method _sample_latents (line 389) | def _sample_latents(self, n):
    method _update_latents (line 393) | def _update_latents(self):
    method _eval_actor (line 408) | def _eval_actor(self, obs, ase_latents):
    method _eval_critic (line 412) | def _eval_critic(self, obs_dict, ase_latents):
    method _calc_amp_rewards (line 422) | def _calc_amp_rewards(self, amp_obs, ase_latents):
    method _calc_enc_rewards (line 431) | def _calc_enc_rewards(self, amp_obs, ase_latents):
    method _enc_loss (line 440) | def _enc_loss(self, enc_pred, ase_latent, enc_obs, loss_mask):
    method _diversity_loss (line 472) | def _diversity_loss(self, obs, action_params, ase_latents):
    method _calc_enc_error (line 496) | def _calc_enc_error(self, enc_pred, ase_latent):
    method _enable_enc_grad_penalty (line 501) | def _enable_enc_grad_penalty(self):
    method _enable_amp_diversity_bonus (line 504) | def _enable_amp_diversity_bonus(self):
    method _eval_enc (line 507) | def _eval_enc(self, amp_obs):
    method _combine_rewards (line 511) | def _combine_rewards(self, task_rewards, amp_rewards):
    method _record_train_batch_info (line 519) | def _record_train_batch_info(self, batch_dict, train_info):
    method _log_train_info (line 524) | def _log_train_info(self, train_info, frame):
    method _change_char_color (line 541) | def _change_char_color(self, env_ids):
    method _amp_debug (line 552) | def _amp_debug(self, info, ase_latents):

FILE: timechamber/ase/ase_models.py
  class ModelASEContinuous (line 31) | class ModelASEContinuous(amp_models.ModelAMPContinuous):
    method __init__ (line 32) | def __init__(self, network):
    method build (line 36) | def build(self, config):
    class Network (line 49) | class Network(amp_models.ModelAMPContinuous.Network):
      method __init__ (line 50) | def __init__(self, a2c_network, obs_shape, normalize_value, normaliz...
      method forward (line 58) | def forward(self, input_dict):
      method eval_actor (line 69) | def eval_actor(self, obs, ase_latents, use_hidden_latents=False):
      method eval_critic (line 74) | def eval_critic(self, obs, ase_latents, use_hidden_latents=False):

FILE: timechamber/ase/ase_network_builder.py
  class LatentType (line 42) | class LatentType(enum.Enum):
  class ASEBuilder (line 46) | class ASEBuilder(amp_network_builder.AMPBuilder):
    method __init__ (line 47) | def __init__(self, **kwargs):
    class Network (line 51) | class Network(amp_network_builder.AMPBuilder.Network):
      method __init__ (line 52) | def __init__(self, params, **kwargs):
      method load (line 120) | def load(self, params):
      method forward (line 130) | def forward(self, obs_dict):
      method eval_critic (line 143) | def eval_critic(self, obs, ase_latents, use_hidden_latents=False):
      method eval_actor (line 151) | def eval_actor(self, obs, ase_latents, use_hidden_latents=False):
      method get_enc_weights (line 174) | def get_enc_weights(self):
      method _build_actor_critic_net (line 183) | def _build_actor_critic_net(self, input_shape, ase_latent_shape):
      method _build_enc (line 213) | def _build_enc(self, input_shape):
      method eval_enc (line 242) | def eval_enc(self, amp_obs):
      method sample_latents (line 249) | def sample_latents(self, n):
    method build (line 255) | def build(self, name, **kwargs):
  class AMPMLPNet (line 260) | class AMPMLPNet(torch.nn.Module):
    method __init__ (line 261) | def __init__(self, obs_size, ase_latent_size, units, activation, initi...
    method forward (line 283) | def forward(self, obs, latent, skip_style):
    method init_params (line 289) | def init_params(self):
    method get_out_size (line 297) | def get_out_size(self):
  class AMPStyleCatNet1 (line 301) | class AMPStyleCatNet1(torch.nn.Module):
    method __init__ (line 302) | def __init__(self, obs_size, ase_latent_size, units, activation,
    method forward (line 333) | def forward(self, obs, latent, skip_style):
    method eval_style (line 348) | def eval_style(self, latent):
    method init_params (line 354) | def init_params(self):
    method get_out_size (line 366) | def get_out_size(self):
    method _build_style_mlp (line 370) | def _build_style_mlp(self, style_units, input_size):

FILE: timechamber/ase/ase_players.py
  class ASEPlayer (line 38) | class ASEPlayer(amp_players.AMPPlayerContinuous):
    method __init__ (line 39) | def __init__(self, params):
    method run (line 58) | def run(self):
    method get_action (line 63) | def get_action(self, obs_dict, is_determenistic=False):
    method env_reset (line 91) | def env_reset(self, env_ids=None):
    method _build_net_config (line 96) | def _build_net_config(self):
    method _reset_latents (line 101) | def _reset_latents(self, done_env_ids=None):
    method _update_latents (line 112) | def _update_latents(self):
    method _reset_latent_step_count (line 126) | def _reset_latent_step_count(self):
    method _calc_amp_rewards (line 130) | def _calc_amp_rewards(self, amp_obs, ase_latents):
    method _calc_enc_rewards (line 139) | def _calc_enc_rewards(self, amp_obs, ase_latents):
    method _calc_enc_error (line 148) | def _calc_enc_error(self, enc_pred, ase_latent):
    method _eval_enc (line 153) | def _eval_enc(self, amp_obs):
    method _amp_debug (line 157) | def _amp_debug(self, info):
    method _change_char_color (line 173) | def _change_char_color(self, env_ids):

FILE: timechamber/ase/hrl_agent.py
  class HRLAgent (line 56) | class HRLAgent(common_agent.CommonAgent):
    method __init__ (line 57) | def __init__(self, base_name, params):
    method env_step (line 75) | def env_step(self, actions):
    method cast_obs (line 117) | def cast_obs(self, obs):
    method preprocess_actions (line 122) | def preprocess_actions(self, actions):
    method play_steps (line 128) | def play_steps(self):
    method _load_config_params (line 198) | def _load_config_params(self, config):
    method _get_mean_rewards (line 205) | def _get_mean_rewards(self):
    method _setup_action_space (line 210) | def _setup_action_space(self):
    method init_tensors (line 215) | def init_tensors(self):
    method _build_llc (line 235) | def _build_llc(self, config_params, checkpoint_file):
    method _build_llc_agent_config (line 243) | def _build_llc_agent_config(self, config_params, network=None):
    method _compute_llc_action (line 259) | def _compute_llc_action(self, obs, actions):
    method _extract_llc_obs (line 270) | def _extract_llc_obs(self, obs):
    method _calc_disc_reward (line 275) | def _calc_disc_reward(self, amp_obs):
    method _combine_rewards (line 279) | def _combine_rewards(self, task_rewards, disc_rewards):
    method _record_train_batch_info (line 286) | def _record_train_batch_info(self, batch_dict, train_info):
    method _log_train_info (line 291) | def _log_train_info(self, train_info, frame):

FILE: timechamber/ase/hrl_models.py
  class ModelHRLContinuous (line 32) | class ModelHRLContinuous(ModelA2CContinuousLogStd):
    method __init__ (line 33) | def __init__(self, network):
    method build (line 37) | def build(self, config):
    class Network (line 49) | class Network(ModelA2CContinuousLogStd.Network):
      method __init__ (line 50) | def __init__(self, a2c_network, obs_shape, normalize_value, normaliz...
      method eval_critic (line 58) | def eval_critic(self, obs):

FILE: timechamber/ase/hrl_network_builder.py
  class HRLBuilder (line 36) | class HRLBuilder(network_builder.A2CBuilder):
    method __init__ (line 37) | def __init__(self, **kwargs):
    class Network (line 41) | class Network(network_builder.A2CBuilder.Network):
      method __init__ (line 42) | def __init__(self, params, **kwargs):
      method forward (line 54) | def forward(self, obs_dict):
      method eval_critic (line 59) | def eval_critic(self, obs):
    method build (line 66) | def build(self, name, **kwargs):

FILE: timechamber/ase/hrl_players.py
  class HRLPlayer (line 47) | class HRLPlayer(common_player.CommonPlayer):
    method __init__ (line 48) | def __init__(self, params):
    method get_action (line 66) | def get_action(self, obs_dict, is_determenistic = False):
    method run (line 92) | def run(self):
    method env_step (line 198) | def env_step(self, env, obs_dict, action):
    method _build_llc (line 238) | def _build_llc(self, config_params, checkpoint_file):
    method _build_llc_agent_config (line 246) | def _build_llc_agent_config(self, config_params, network=None):
    method _setup_action_space (line 261) | def _setup_action_space(self):
    method _compute_llc_action (line 266) | def _compute_llc_action(self, obs, actions):
    method _extract_llc_obs (line 276) | def _extract_llc_obs(self, obs):
    method _calc_disc_reward (line 281) | def _calc_disc_reward(self, amp_obs):

FILE: timechamber/ase/utils/amp_agent.py
  class AMPAgent (line 49) | class AMPAgent(common_agent.CommonAgent):
    method __init__ (line 50) | def __init__(self, base_name, params):
    method init_tensors (line 58) | def init_tensors(self):
    method set_eval (line 63) | def set_eval(self):
    method set_train (line 69) | def set_train(self):
    method get_stats_weights (line 75) | def get_stats_weights(self):
    method set_stats_weights (line 82) | def set_stats_weights(self, weights):
    method play_steps (line 89) | def play_steps(self):
    method get_action_values (line 167) | def get_action_values(self, obs_dict, rand_action_probs):
    method prepare_dataset (line 199) | def prepare_dataset(self, batch_dict):
    method train_epoch (line 209) | def train_epoch(self):
    method calc_gradients (line 294) | def calc_gradients(self, input_dict):
    method _load_config_params (line 420) | def _load_config_params(self, config):
    method _build_net_config (line 447) | def _build_net_config(self):
    method _build_rand_action_probs (line 452) | def _build_rand_action_probs(self):
    method _init_train (line 465) | def _init_train(self):
    method _disc_loss (line 470) | def _disc_loss(self, disc_agent_logit, disc_demo_logit, obs_demo):
    method _disc_loss_neg (line 509) | def _disc_loss_neg(self, disc_logits):
    method _disc_loss_pos (line 514) | def _disc_loss_pos(self, disc_logits):
    method _compute_disc_acc (line 519) | def _compute_disc_acc(self, disc_agent_logit, disc_demo_logit):
    method _fetch_amp_obs_demo (line 526) | def _fetch_amp_obs_demo(self, num_samples):
    method _build_amp_buffers (line 530) | def _build_amp_buffers(self):
    method _init_amp_demo_buf (line 548) | def _init_amp_demo_buf(self):
    method _update_amp_demos (line 558) | def _update_amp_demos(self):
    method _preproc_amp_obs (line 563) | def _preproc_amp_obs(self, amp_obs):
    method _combine_rewards (line 568) | def _combine_rewards(self, task_rewards, amp_rewards):
    method _eval_disc (line 575) | def _eval_disc(self, amp_obs):
    method _calc_advs (line 579) | def _calc_advs(self, batch_dict):
    method _calc_amp_rewards (line 591) | def _calc_amp_rewards(self, amp_obs):
    method _calc_disc_rewards (line 598) | def _calc_disc_rewards(self, amp_obs):
    method _store_replay_amp_obs (line 607) | def _store_replay_amp_obs(self, amp_obs):
    method _record_train_batch_info (line 624) | def _record_train_batch_info(self, batch_dict, train_info):
    method _log_train_info (line 629) | def _log_train_info(self, train_info, frame):
    method _amp_debug (line 646) | def _amp_debug(self, info):

FILE: timechamber/ase/utils/amp_datasets.py
  class AMPDataset (line 32) | class AMPDataset(datasets.PPODataset):
    method __init__ (line 33) | def __init__(self, batch_size, minibatch_size, is_discrete, is_rnn, de...
    method update_mu_sigma (line 38) | def update_mu_sigma(self, mu, sigma):
    method _get_item (line 42) | def _get_item(self, idx):
    method _shuffle_idx_buf (line 57) | def _shuffle_idx_buf(self):

FILE: timechamber/ase/utils/amp_models.py
  class ModelAMPContinuous (line 33) | class ModelAMPContinuous(ModelA2CContinuousLogStd):
    method __init__ (line 34) | def __init__(self, network):
    method build (line 38) | def build(self, config):
    class Network (line 51) | class Network(ModelA2CContinuousLogStd.Network):
      method __init__ (line 52) | def __init__(self, a2c_network, obs_shape, normalize_value, normaliz...
      method forward (line 59) | def forward(self, input_dict):
      method eval_actor (line 78) | def eval_actor(self, obs):
      method eval_critic (line 83) | def eval_critic(self, obs):

FILE: timechamber/ase/utils/amp_network_builder.py
  class AMPBuilder (line 39) | class AMPBuilder(network_builder.A2CBuilder):
    method __init__ (line 40) | def __init__(self, **kwargs):
    class Network (line 44) | class Network(network_builder.A2CBuilder.Network):
      method __init__ (line 45) | def __init__(self, params, **kwargs):
      method load (line 60) | def load(self, params):
      method forward (line 68) | def forward(self, obs_dict):
      method eval_actor (line 79) | def eval_actor(self, obs):
      method eval_critic (line 102) | def eval_critic(self, obs):
      method eval_disc (line 109) | def eval_disc(self, amp_obs):
      method get_disc_logit_weights (line 114) | def get_disc_logit_weights(self):
      method get_disc_weights (line 117) | def get_disc_weights(self):
      method _build_disc (line 126) | def _build_disc(self, input_shape):
    method build (line 152) | def build(self, name, **kwargs):

FILE: timechamber/ase/utils/amp_players.py
  class AMPPlayerContinuous (line 36) | class AMPPlayerContinuous(common_player.CommonPlayer):
    method __init__ (line 37) | def __init__(self, params):
    method restore (line 45) | def restore(self, fn):
    method _build_net (line 53) | def _build_net(self, config):
    method _post_step (line 62) | def _post_step(self, info):
    method _build_net_config (line 68) | def _build_net_config(self):
    method _amp_debug (line 76) | def _amp_debug(self, info):
    method _preproc_amp_obs (line 90) | def _preproc_amp_obs(self, amp_obs):
    method _eval_disc (line 95) | def _eval_disc(self, amp_obs):
    method _calc_amp_rewards (line 99) | def _calc_amp_rewards(self, amp_obs):
    method _calc_disc_rewards (line 106) | def _calc_disc_rewards(self, amp_obs):

FILE: timechamber/ase/utils/common_agent.py
  class CommonAgent (line 54) | class CommonAgent(a2c_continuous.A2CAgent):
    method __init__ (line 55) | def __init__(self, base_name, params):
    method init_tensors (line 105) | def init_tensors(self):
    method train (line 113) | def train(self):
    method set_full_state_weights (line 188) | def set_full_state_weights(self, weights):
    method restore (line 203) | def restore(self, fn):
    method train_epoch (line 210) | def train_epoch(self):
    method play_steps (line 283) | def play_steps(self):
    method prepare_dataset (line 348) | def prepare_dataset(self, batch_dict):
    method calc_gradients (line 394) | def calc_gradients(self, input_dict):
    method discount_values (line 478) | def discount_values(self, mb_fdones, mb_values, mb_rewards, mb_next_va...
    method env_reset (line 492) | def env_reset(self, env_ids=None):
    method bound_loss (line 497) | def bound_loss(self, mu):
    method _get_mean_rewards (line 507) | def _get_mean_rewards(self):
    method _load_config_params (line 510) | def _load_config_params(self, config):
    method _build_net_config (line 514) | def _build_net_config(self):
    method _setup_action_space (line 526) | def _setup_action_space(self):
    method _init_train (line 535) | def _init_train(self):
    method _eval_critic (line 538) | def _eval_critic(self, obs_dict):
    method _actor_loss (line 546) | def _actor_loss(self, old_action_log_probs_batch, action_log_probs, ad...
    method _critic_loss (line 562) | def _critic_loss(self, value_preds_batch, values, curr_e_clip, return_...
    method _calc_advs (line 577) | def _calc_advs(self, batch_dict):
    method _record_train_batch_info (line 589) | def _record_train_batch_info(self, batch_dict, train_info):
    method _log_train_info (line 592) | def _log_train_info(self, train_info, frame):

FILE: timechamber/ase/utils/common_player.py
  class CommonPlayer (line 39) | class CommonPlayer(players.PpoPlayerContinuous):
    method __init__ (line 40) | def __init__(self, params):
    method run (line 56) | def run(self):
    method get_action (line 162) | def get_action(self, obs_dict, is_determenistic = False):
    method env_step (line 166) | def env_step(self, env, actions):
    method _build_net (line 183) | def _build_net(self, config):
    method env_reset (line 194) | def env_reset(self, env_ids=None):
    method _post_step (line 198) | def _post_step(self, info):
    method _build_net_config (line 201) | def _build_net_config(self):
    method restore (line 212) | def restore(self, fn):
    method _setup_action_space (line 222) | def _setup_action_space(self):

FILE: timechamber/ase/utils/replay_buffer.py
  class ReplayBuffer (line 31) | class ReplayBuffer():
    method __init__ (line 32) | def __init__(self, buffer_size, device):
    method reset (line 43) | def reset(self):
    method get_buffer_size (line 49) | def get_buffer_size(self):
    method get_total_count (line 52) | def get_total_count(self):
    method store (line 55) | def store(self, data_dict):
    method sample (line 79) | def sample(self, n):
    method _reset_sample_idx (line 99) | def _reset_sample_idx(self):
    method _init_data_buf (line 105) | def _init_data_buf(self, data_dict):

FILE: timechamber/learning/common_agent.py
  class CommonAgent (line 26) | class CommonAgent(a2c_continuous.A2CAgent):
    method __init__ (line 28) | def __init__(self, base_name, params):
    method init_tensors (line 76) | def init_tensors(self):
    method train (line 84) | def train(self):
    method train_epoch (line 155) | def train_epoch(self):
    method play_steps (line 228) | def play_steps(self):
    method calc_gradients (line 290) | def calc_gradients(self, input_dict):
    method discount_values (line 384) | def discount_values(self, mb_fdones, mb_values, mb_rewards, mb_next_va...
    method bound_loss (line 398) | def bound_loss(self, mu):
    method _load_config_params (line 408) | def _load_config_params(self, config):
    method _build_net_config (line 412) | def _build_net_config(self):
    method _setup_action_space (line 424) | def _setup_action_space(self):
    method _init_train (line 433) | def _init_train(self):
    method _env_reset_done (line 436) | def _env_reset_done(self):
    method _eval_critic (line 440) | def _eval_critic(self, obs_dict):
    method _actor_loss (line 453) | def _actor_loss(self, old_action_log_probs_batch, action_log_probs, ad...
    method _critic_loss (line 474) | def _critic_loss(self, value_preds_batch, values, curr_e_clip, return_...
    method _record_train_batch_info (line 489) | def _record_train_batch_info(self, batch_dict, train_info):
    method _log_train_info (line 492) | def _log_train_info(self, train_info, frame):

FILE: timechamber/learning/common_player.py
  class CommonPlayer (line 11) | class CommonPlayer(players.PpoPlayerContinuous):
    method __init__ (line 13) | def __init__(self, params):
    method run (line 28) | def run(self):
    method obs_to_torch (line 128) | def obs_to_torch(self, obs):
    method get_action (line 135) | def get_action(self, obs_dict, is_determenistic = False):
    method _build_net (line 139) | def _build_net(self, config):
    method _env_reset_done (line 147) | def _env_reset_done(self):
    method _post_step (line 151) | def _post_step(self, info):
    method _build_net_config (line 154) | def _build_net_config(self):
    method _setup_action_space (line 166) | def _setup_action_space(self):

FILE: timechamber/learning/hrl_sp_agent.py
  class HRLSPAgent (line 21) | class HRLSPAgent(hrl_agent.HRLAgent):
    method __init__ (line 22) | def __init__(self, base_name, params):
    method _build_player_pool (line 56) | def _build_player_pool(self, params):
    method play_steps (line 67) | def play_steps(self):
    method env_step (line 153) | def env_step(self, ego_actions, op_actions):
    method env_reset (line 223) | def env_reset(self, env_ids=None):
    method train (line 231) | def train(self):
    method update_metric (line 344) | def update_metric(self):
    method get_action_values (line 360) | def get_action_values(self, obs, is_op=False):
    method restore (line 390) | def restore(self, fn):
    method resample_op (line 397) | def resample_op(self, resample_indices):
    method resample_batch (line 408) | def resample_batch(self):
    method restore_op (line 420) | def restore_op(self, fn):
    method check_update_opponent (line 428) | def check_update_opponent(self, win_rate):
    method create_model (line 438) | def create_model(self):
    method update_player_pool (line 443) | def update_player_pool(self, model, player_idx):

FILE: timechamber/learning/hrl_sp_player.py
  class HRLSPPlayer (line 19) | class HRLSPPlayer(hrl_players.HRLPlayer):
    method __init__ (line 20) | def __init__(self, params):
    method restore (line 52) | def restore(self, load_dir):
    method restore_op (line 80) | def restore_op(self, load_dir):
    method _alloc_env_indices (line 104) | def _alloc_env_indices(self):
    method _build_player_pool (line 120) | def _build_player_pool(self, params, player_num):
    method _update_rating (line 138) | def _update_rating(self, info, env_indices):
    method run (line 159) | def run(self):
    method _plot_elo_curve (line 253) | def _plot_elo_curve(self):
    method get_action (line 277) | def get_action(self, obs, is_determenistic=False, is_op=False):
    method _norm_policy_timestep (line 308) | def _norm_policy_timestep(self):
    method env_reset (line 321) | def env_reset(self, env, env_ids=None):
    method env_step (line 328) | def env_step(self, env, obs_dict, ego_actions, op_actions):
    method create_model (line 396) | def create_model(self):
    method load_model (line 401) | def load_model(self, fn):

FILE: timechamber/learning/pfsp_player_pool.py
  function player_inference_thread (line 12) | def player_inference_thread(model, input_dict, res_dict, env_indices, pr...
  function player_inference_process (line 22) | def player_inference_process(pipe, queue, barrier):
  class SinglePlayer (line 54) | class SinglePlayer:
    method __init__ (line 55) | def __init__(self, player_idx, model, device, obs_batch_len=0, rating=...
    method __call__ (line 71) | def __call__(self, input_dict):
    method reset_envs (line 74) | def reset_envs(self):
    method remove_envs (line 77) | def remove_envs(self, env_indices):
    method add_envs (line 80) | def add_envs(self, env_indices):
    method clear_envs (line 83) | def clear_envs(self):
    method update_metric (line 86) | def update_metric(self, wins, loses, draws):
    method clear_metric (line 97) | def clear_metric(self):
    method win_rate (line 103) | def win_rate(self):
    method games_num (line 110) | def games_num(self):
  class PFSPPlayerPool (line 114) | class PFSPPlayerPool:
    method __init__ (line 115) | def __init__(self, max_length, device):
    method add_player (line 127) | def add_player(self, player):
    method sample_player (line 135) | def sample_player(self, weight='linear'):
    method update_player_metric (line 141) | def update_player_metric(self, infos):
    method clear_player_metric (line 145) | def clear_player_metric(self):
    method inference (line 149) | def inference(self, input_dict, res_dict, processed_obs):
  class PFSPPlayerVectorizedPool (line 159) | class PFSPPlayerVectorizedPool(PFSPPlayerPool):
    method __init__ (line 160) | def __init__(self, max_length, device, vector_model_config, params):
    method inference (line 175) | def inference(self, input_dict, res_dict, processed_obs):
    method add_player (line 186) | def add_player(self, player):
  class PFSPPlayerThreadPool (line 192) | class PFSPPlayerThreadPool(PFSPPlayerPool):
    method __init__ (line 193) | def __init__(self, max_length, device):
    method inference (line 197) | def inference(self, input_dict, res_dict, processed_obs):
  class PFSPPlayerProcessPool (line 205) | class PFSPPlayerProcessPool(PFSPPlayerPool):
    method __init__ (line 206) | def __init__(self, max_length, device):
    method _init_inference_processes (line 216) | def _init_inference_processes(self):
    method add_player (line 230) | def add_player(self, player):
    method inference (line 247) | def inference(self, input_dict, res_dict, processed_obs):
    method __del__ (line 258) | def __del__(self):

FILE: timechamber/learning/ppo_sp_agent.py
  class SPAgent (line 22) | class SPAgent(a2c_continuous.A2CAgent):
    method __init__ (line 23) | def __init__(self, base_name, params):
    method _build_player_pool (line 57) | def _build_player_pool(self, params):
    method play_steps (line 74) | def play_steps(self):
    method env_step (line 145) | def env_step(self, actions):
    method env_reset (line 160) | def env_reset(self, env_ids=None):
    method train (line 167) | def train(self):
    method update_metric (line 266) | def update_metric(self):
    method get_action_values (line 282) | def get_action_values(self, obs, is_op=False):
    method resample_op (line 312) | def resample_op(self, resample_indices):
    method resample_batch (line 323) | def resample_batch(self):
    method restore_op (line 335) | def restore_op(self, fn):
    method check_update_opponent (line 341) | def check_update_opponent(self, win_rate):
    method create_model (line 351) | def create_model(self):
    method update_player_pool (line 356) | def update_player_pool(self, model, player_idx):

FILE: timechamber/learning/ppo_sp_player.py
  function rescale_actions (line 18) | def rescale_actions(low, high, action):
  class SPPlayer (line 25) | class SPPlayer(BasePlayer):
    method __init__ (line 26) | def __init__(self, params):
    method restore (line 61) | def restore(self, load_dir):
    method restore_op (line 89) | def restore_op(self, load_dir):
    method _alloc_env_indices (line 113) | def _alloc_env_indices(self):
    method _build_player_pool (line 129) | def _build_player_pool(self, params, player_num):
    method _update_rating (line 147) | def _update_rating(self, info, env_indices):
    method run (line 170) | def run(self):
    method _plot_elo_curve (line 265) | def _plot_elo_curve(self):
    method get_action (line 292) | def get_action(self, obs, is_determenistic=False, is_op=False):
    method _norm_policy_timestep (line 328) | def _norm_policy_timestep(self):
    method env_reset (line 341) | def env_reset(self, env, done_indices=None):
    method env_step (line 348) | def env_step(self, env, actions):
    method create_model (line 365) | def create_model(self):
    method load_model (line 370) | def load_model(self, fn):

FILE: timechamber/learning/replay_buffer.py
  class ReplayBuffer (line 6) | class ReplayBuffer():
    method __init__ (line 7) | def __init__(self, buffer_size, device):
    method reset (line 18) | def reset(self):
    method get_buffer_size (line 24) | def get_buffer_size(self):
    method get_total_count (line 27) | def get_total_count(self):
    method store (line 30) | def store(self, data_dict):
    method sample (line 54) | def sample(self, n):
    method _reset_sample_idx (line 74) | def _reset_sample_idx(self):
    method _init_data_buf (line 80) | def _init_data_buf(self, data_dict):

FILE: timechamber/learning/vectorized_models.py
  class VectorizedRunningMeanStd (line 8) | class VectorizedRunningMeanStd(RunningMeanStd):
    method __init__ (line 9) | def __init__(self, insize, population_size, epsilon=1e-05, per_channel...
    method _update_mean_var_count_from_moments (line 34) | def _update_mean_var_count_from_moments(self, mean, var, count, batch_...
    method forward (line 45) | def forward(self, input, unnorm=False, mask=None):
  class ModelVectorizedA2C (line 86) | class ModelVectorizedA2C(ModelA2CContinuousLogStd):
    method __init__ (line 87) | def __init__(self, network):
    method build (line 91) | def build(self, config):
    class Network (line 105) | class Network(ModelA2CContinuousLogStd.Network):
      method __init__ (line 106) | def __init__(self, a2c_network, population_size, obs_shape, normaliz...
      method update (line 118) | def update(self, population_idx, network):

FILE: timechamber/learning/vectorized_network_builder.py
  class VectorizedLinearLayer (line 7) | class VectorizedLinearLayer(torch.nn.Module):
    method __init__ (line 10) | def __init__(
    method forward (line 43) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class VectorizedA2CBuilder (line 50) | class VectorizedA2CBuilder(network_builder.A2CBuilder):
    method __init__ (line 51) | def __init__(self, **kwargs):
    class Network (line 55) | class Network(network_builder.A2CBuilder.Network):
      method __init__ (line 56) | def __init__(self, params, **kwargs):
      method _build_vectorized_mlp (line 72) | def _build_vectorized_mlp(self,
      method _build_mlp (line 87) | def _build_mlp(self,
      method forward (line 97) | def forward(self, obs_dict):  # implement continues situation
      method load (line 109) | def load(self, params):
    method build (line 112) | def build(self, name, **kwargs):

FILE: timechamber/tasks/ase_humanoid_base/base_task.py
  class BaseTask (line 22) | class BaseTask():
    method __init__ (line 24) | def __init__(self, cfg, enable_camera_sensors=False):
    method set_sim_params_up_axis (line 110) | def set_sim_params_up_axis(self, sim_params, axis):
    method create_sim (line 119) | def create_sim(self, compute_device, graphics_device, physics_engine, ...
    method step (line 127) | def step(self, actions):
    method get_states (line 147) | def get_states(self):
    method render (line 150) | def render(self, sync_frame_time=False):
    method get_actor_params_info (line 174) | def get_actor_params_info(self, dr_params, env):
    method apply_randomizations (line 212) | def apply_randomizations(self, dr_params):
    method pre_physics_step (line 408) | def pre_physics_step(self, actions):
    method _physics_step (line 411) | def _physics_step(self):
    method post_physics_step (line 417) | def post_physics_step(self):
  function get_attr_val_from_sample (line 421) | def get_attr_val_from_sample(sample, offset, prop, attr):

FILE: timechamber/tasks/ase_humanoid_base/humanoid.py
  class Humanoid (line 41) | class Humanoid(BaseTask):
    method __init__ (line 42) | def __init__(self, cfg, sim_params, physics_engine, device_type, devic...
    method get_obs_size (line 157) | def get_obs_size(self):
    method get_action_size (line 160) | def get_action_size(self):
    method get_num_actors_per_env (line 163) | def get_num_actors_per_env(self):
    method _add_circle_borderline (line 167) | def _add_circle_borderline(self, env):
    method _add_rectangle_borderline (line 180) | def _add_rectangle_borderline(self, env):
    method allocate_buffers (line 232) | def allocate_buffers(self):
    method create_sim (line 254) | def create_sim(self):
    method reset (line 262) | def reset(self, env_ids=None):
    method set_char_color (line 268) | def set_char_color(self, col, env_ids):
    method _reset_envs (line 279) | def _reset_envs(self, env_ids):
    method _reset_env_tensors (line 287) | def _reset_env_tensors(self, env_ids):
    method _create_ground_plane (line 304) | def _create_ground_plane(self):
    method _setup_character_props (line 313) | def _setup_character_props(self, key_bodies):
    method _build_termination_heights (line 337) | def _build_termination_heights(self):
    method _create_envs (line 355) | def _create_envs(self, num_envs, spacing, num_per_row):
    method _build_env (line 442) | def _build_env(self, env_id, env_ptr, humanoid_asset, humanoid_asset_op):
    method _build_pd_action_offset_scale (line 488) | def _build_pd_action_offset_scale(self):
    method _get_humanoid_collision_filter (line 534) | def _get_humanoid_collision_filter(self):
    method _compute_reward (line 537) | def _compute_reward(self, actions):
    method _compute_reset (line 541) | def _compute_reset(self):
    method _refresh_sim_tensors (line 548) | def _refresh_sim_tensors(self):
    method _compute_observations (line 558) | def _compute_observations(self):
    method _compute_humanoid_obs (line 566) | def _compute_humanoid_obs(self):
    method _reset_actors (line 585) | def _reset_actors(self, env_ids):
    method pre_physics_step (line 594) | def pre_physics_step(self, actions):
    method post_physics_step (line 612) | def post_physics_step(self):
    method render (line 628) | def render(self, sync_frame_time=False):
    method _build_key_body_ids_tensor (line 633) | def _build_key_body_ids_tensor(self, key_body_names):
    method _build_contact_body_ids_tensor (line 646) | def _build_contact_body_ids_tensor(self, contact_body_names):
    method _action_to_pd_targets (line 659) | def _action_to_pd_targets(self, action):
    method _update_debug_viz (line 663) | def _update_debug_viz(self):
  function dof_to_obs (line 672) | def dof_to_obs(pose, dof_obs_size, dof_offsets):
  function compute_humanoid_observations (line 704) | def compute_humanoid_observations(root_pos, root_rot, root_vel, root_ang...
  function compute_humanoid_observations_max (line 741) | def compute_humanoid_observations_max(body_pos, body_rot, body_vel, body...
  function expand_env_ids (line 789) | def expand_env_ids(env_ids, n_agents):
  function compute_humanoid_reward (line 798) | def compute_humanoid_reward(obs_buf):
  function compute_humanoid_reset (line 804) | def compute_humanoid_reset(reset_buf, progress_buf, contact_buf, contact...

FILE: timechamber/tasks/ase_humanoid_base/humanoid_amp.py
  class HumanoidAMP (line 43) | class HumanoidAMP(Humanoid):
    class StateInit (line 44) | class StateInit(Enum):
    method __init__ (line 50) | def __init__(self, cfg, sim_params, physics_engine, device_type, devic...
    method post_physics_step (line 78) | def post_physics_step(self):
    method get_num_amp_obs (line 89) | def get_num_amp_obs(self):
    method fetch_amp_obs_demo (line 92) | def fetch_amp_obs_demo(self, num_samples):
    method build_amp_obs_demo (line 107) | def build_amp_obs_demo(self, motion_ids, motion_times0):
    method _build_amp_obs_demo_buf (line 125) | def _build_amp_obs_demo_buf(self, num_samples):
    method _setup_character_props (line 129) | def _setup_character_props(self, key_bodies):
    method _load_motion (line 145) | def _load_motion(self, motion_file):
    method _reset_envs (line 154) | def _reset_envs(self, env_ids):
    method _reset_actors (line 162) | def _reset_actors(self, env_ids):
    method _reset_default (line 174) | def _reset_default(self, env_ids):
    method _reset_ref_state_init (line 182) | def _reset_ref_state_init(self, env_ids):
    method _reset_hybrid_state_init (line 210) | def _reset_hybrid_state_init(self, env_ids):
    method _init_amp_obs (line 225) | def _init_amp_obs(self, env_ids):
    method _init_amp_obs_default (line 237) | def _init_amp_obs_default(self, env_ids):
    method _init_amp_obs_ref (line 242) | def _init_amp_obs_ref(self, env_ids, motion_ids, motion_times):
    method _set_env_state (line 260) | def _set_env_state(self, env_ids, root_pos, root_rot, dof_pos, root_ve...
    method _update_hist_amp_obs (line 270) | def _update_hist_amp_obs(self, env_ids=None):
    method _compute_amp_observations (line 277) | def _compute_amp_observations(self, env_ids=None):
  function build_amp_observations (line 303) | def build_amp_observations(root_pos, root_rot, root_vel, root_ang_vel, d...

FILE: timechamber/tasks/ase_humanoid_base/humanoid_amp_task.py
  class HumanoidAMPTask (line 33) | class HumanoidAMPTask(humanoid_amp.HumanoidAMP):
    method __init__ (line 34) | def __init__(self, cfg, sim_params, physics_engine, device_type, devic...
    method get_obs_size (line 45) | def get_obs_size(self):
    method get_task_obs_size (line 52) | def get_task_obs_size(self):
    method pre_physics_step (line 55) | def pre_physics_step(self, actions):
    method render (line 60) | def render(self, sync_frame_time=False):
    method _update_task (line 67) | def _update_task(self):
    method _reset_envs (line 70) | def _reset_envs(self, env_ids):
    method _reset_task (line 75) | def _reset_task(self, env_ids):
    method _compute_observations (line 78) | def _compute_observations(self):
    method _compute_task_obs (line 102) | def _compute_task_obs(self, env_ids=None):
    method _compute_reward (line 105) | def _compute_reward(self, actions):
    method _draw_task (line 108) | def _draw_task(self):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/core/backend/abstract.py
  function register (line 40) | def register(name):
  function _get_cls (line 50) | def _get_cls(name):
  class NumpyEncoder (line 55) | class NumpyEncoder(json.JSONEncoder):
    method default (line 58) | def default(self, obj):
  function json_numpy_obj_hook (line 83) | def json_numpy_obj_hook(dct):
  class Serializable (line 90) | class Serializable:
    method from_dict (line 97) | def from_dict(cls, dict_repr, *args, **kwargs):
    method to_dict (line 108) | def to_dict(self):
    method from_file (line 116) | def from_file(cls, path, *args, **kwargs):
    method to_file (line 136) | def to_file(self, path: str) -> None:

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/core/rotation3d.py
  function quat_mul (line 37) | def quat_mul(a, b):
  function quat_pos (line 53) | def quat_pos(x):
  function quat_abs (line 64) | def quat_abs(x):
  function quat_unit (line 73) | def quat_unit(x):
  function quat_conjugate (line 82) | def quat_conjugate(x):
  function quat_real (line 90) | def quat_real(x):
  function quat_imaginary (line 98) | def quat_imaginary(x):
  function quat_norm_check (line 106) | def quat_norm_check(x):
  function quat_normalize (line 117) | def quat_normalize(q):
  function quat_from_xyz (line 126) | def quat_from_xyz(xyz):
  function quat_identity (line 136) | def quat_identity(shape: List[int]):
  function quat_from_angle_axis (line 147) | def quat_from_angle_axis(angle, axis, degree: bool = False):
  function quat_from_rotation_matrix (line 171) | def quat_from_rotation_matrix(m):
  function quat_mul_norm (line 221) | def quat_mul_norm(x, y):
  function quat_rotate (line 230) | def quat_rotate(rot, vec):
  function quat_inverse (line 239) | def quat_inverse(x):
  function quat_identity_like (line 247) | def quat_identity_like(x):
  function quat_angle_axis (line 255) | def quat_angle_axis(x):
  function quat_yaw_rotation (line 268) | def quat_yaw_rotation(x, z_up: bool = True):
  function transform_from_rotation_translation (line 289) | def transform_from_rotation_translation(
  function transform_identity (line 305) | def transform_identity(shape: List[int]):
  function transform_rotation (line 316) | def transform_rotation(x):
  function transform_translation (line 322) | def transform_translation(x):
  function transform_inverse (line 328) | def transform_inverse(x):
  function transform_identity_like (line 339) | def transform_identity_like(x):
  function transform_mul (line 347) | def transform_mul(x, y):
  function transform_apply (line 360) | def transform_apply(rot, vec):
  function rot_matrix_det (line 369) | def rot_matrix_det(x):
  function rot_matrix_integrity_check (line 384) | def rot_matrix_integrity_check(x):
  function rot_matrix_from_quaternion (line 399) | def rot_matrix_from_quaternion(q):
  function euclidean_to_rotation_matrix (line 427) | def euclidean_to_rotation_matrix(x):
  function euclidean_integrity_check (line 435) | def euclidean_integrity_check(x):
  function euclidean_translation (line 442) | def euclidean_translation(x):
  function euclidean_inverse (line 450) | def euclidean_inverse(x):
  function euclidean_to_transform (line 462) | def euclidean_to_transform(transformation_matrix):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/core/tensor_utils.py
  class TensorUtils (line 13) | class TensorUtils(Serializable):
    method from_dict (line 15) | def from_dict(cls, dict_repr, *args, **kwargs):
    method to_dict (line 25) | def to_dict(self):
  function tensor_to_dict (line 32) | def tensor_to_dict(x):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/skeleton/backend/fbx/fbx_backend.py
  function fbx_to_npy (line 48) | def fbx_to_npy(file_name_in, root_joint_name, fps):
  function _get_frame_count (line 152) | def _get_frame_count(fbx_scene):
  function _get_animation_curve (line 177) | def _get_animation_curve(joint, fbx_scene):
  function _get_skeleton (line 233) | def _get_skeleton(root_joint):
  function _recursive_to_list (line 258) | def _recursive_to_list(array):
  function parse_fbx (line 273) | def parse_fbx(file_name_in, root_joint_name, fps):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/skeleton/backend/fbx/fbx_read_wrapper.py
  function fbx_to_array (line 25) | def fbx_to_array(fbx_file_path, root_joint, fps):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/skeleton/skeleton3d.py
  class SkeletonTree (line 42) | class SkeletonTree(Serializable):
    method __init__ (line 99) | def __init__(self, node_names, parent_indices, local_translation):
    method __len__ (line 116) | def __len__(self):
    method __iter__ (line 120) | def __iter__(self):
    method __getitem__ (line 124) | def __getitem__(self, item):
    method __repr__ (line 128) | def __repr__(self):
    method _indent (line 138) | def _indent(self, s):
    method node_names (line 142) | def node_names(self):
    method parent_indices (line 146) | def parent_indices(self):
    method local_translation (line 150) | def local_translation(self):
    method num_joints (line 154) | def num_joints(self):
    method from_dict (line 159) | def from_dict(cls, dict_repr, *args, **kwargs):
    method to_dict (line 166) | def to_dict(self):
    method from_mjcf (line 176) | def from_mjcf(cls, path: str) -> "SkeletonTree":
    method parent_of (line 222) | def parent_of(self, node_name):
    method index (line 231) | def index(self, node_name):
    method drop_nodes_by_names (line 240) | def drop_nodes_by_names(
    method keep_nodes_by_names (line 283) | def keep_nodes_by_names(
  class SkeletonState (line 290) | class SkeletonState(Serializable):
    method __init__ (line 360) | def __init__(self, tensor_backend, skeleton_tree, is_local):
    method __len__ (line 365) | def __len__(self):
    method rotation (line 369) | def rotation(self):
    method _local_rotation (line 377) | def _local_rotation(self):
    method _global_rotation (line 384) | def _global_rotation(self):
    method is_local (line 391) | def is_local(self):
    method invariant_property (line 399) | def invariant_property(self):
    method num_joints (line 403) | def num_joints(self):
    method skeleton_tree (line 411) | def skeleton_tree(self):
    method root_translation (line 419) | def root_translation(self):
    method global_transformation (line 431) | def global_transformation(self):
    method global_rotation (line 455) | def global_rotation(self):
    method global_translation (line 468) | def global_translation(self):
    method global_translation_xy (line 475) | def global_translation_xy(self):
    method global_translation_xz (line 482) | def global_translation_xz(self):
    method local_rotation (line 490) | def local_rotation(self):
    method local_transformation (line 513) | def local_transformation(self):
    method local_translation (line 523) | def local_translation(self):
    method root_translation_xy (line 542) | def root_translation_xy(self):
    method global_root_rotation (line 549) | def global_root_rotation(self):
    method global_root_yaw_rotation (line 556) | def global_root_yaw_rotation(self):
    method local_translation_to_root (line 564) | def local_translation_to_root(self):
    method local_rotation_to_root (line 573) | def local_rotation_to_root(self):
    method compute_forward_vector (line 580) | def compute_forward_vector(
    method _to_state_vector (line 620) | def _to_state_vector(rot, rt):
    method from_dict (line 630) | def from_dict(
    method to_dict (line 641) | def to_dict(self) -> OrderedDict:
    method from_rotation_and_root_translation (line 652) | def from_rotation_and_root_translation(cls, skeleton_tree, r, t, is_lo...
    method zero_pose (line 675) | def zero_pose(cls, skeleton_tree):
    method local_repr (line 690) | def local_repr(self):
    method global_repr (line 706) | def global_repr(self):
    method _get_pairwise_average_translation (line 722) | def _get_pairwise_average_translation(self):
    method _transfer_to (line 734) | def _transfer_to(self, new_skeleton_tree: SkeletonTree):
    method drop_nodes_by_names (line 743) | def drop_nodes_by_names(
    method keep_nodes_by_names (line 766) | def keep_nodes_by_names(
    method _remapped_to (line 785) | def _remapped_to(
    method retarget_to (line 814) | def retarget_to(
    method retarget_to_by_tpose (line 978) | def retarget_to_by_tpose(
  class SkeletonMotion (line 1026) | class SkeletonMotion(SkeletonState):
    method __init__ (line 1027) | def __init__(self, tensor_backend, skeleton_tree, is_local, fps, *args...
    method clone (line 1031) | def clone(self):
    method invariant_property (line 1037) | def invariant_property(self):
    method global_velocity (line 1045) | def global_velocity(self):
    method global_angular_velocity (line 1053) | def global_angular_velocity(self):
    method fps (line 1061) | def fps(self):
    method time_delta (line 1066) | def time_delta(self):
    method global_root_velocity (line 1071) | def global_root_velocity(self):
    method global_root_angular_velocity (line 1076) | def global_root_angular_velocity(self):
    method from_state_vector_and_velocity (line 1081) | def from_state_vector_and_velocity(
    method from_skeleton_state (line 1118) | def from_skeleton_state(
    method _to_state_vector (line 1151) | def _to_state_vector(rot, rt, vel, avel):
    method from_dict (line 1160) | def from_dict(
    method to_dict (line 1178) | def to_dict(self) -> OrderedDict:
    method from_fbx (line 1192) | def from_fbx(
    method _compute_velocity (line 1251) | def _compute_velocity(p, time_delta, guassian_filter=True):
    method _compute_angular_velocity (line 1261) | def _compute_angular_velocity(r, time_delta: float, guassian_filter=Tr...
    method crop (line 1276) | def crop(self, start: int, end: int, fps: Optional[int] = None):
    method retarget_to (line 1311) | def retarget_to(
    method retarget_to_by_tpose (line 1373) | def retarget_to_by_tpose(

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/visualization/common.py
  function plot_skeleton_state (line 36) | def plot_skeleton_state(skeleton_state, task_name=""):
  function plot_skeleton_states (line 51) | def plot_skeleton_states(skeleton_state, skip_n=1, task_name=""):
  function plot_skeleton_motion (line 72) | def plot_skeleton_motion(skeleton_motion, skip_n=1, task_name=""):
  function plot_skeleton_motion_interactive_base (line 94) | def plot_skeleton_motion_interactive_base(skeleton_motion, task_name=""):
  function plot_skeleton_motion_interactive (line 189) | def plot_skeleton_motion_interactive(skeleton_motion, task_name=""):
  function plot_skeleton_motion_interactive_multiple (line 202) | def plot_skeleton_motion_interactive_multiple(*callables, sync=True):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/visualization/core.py
  class BasePlotterTask (line 36) | class BasePlotterTask(object):
    method __init__ (line 40) | def __init__(self, task_name: str, task_type: str) -> None:
    method task_name (line 45) | def task_name(self):
    method task_type (line 49) | def task_type(self):
    method get_scoped_name (line 52) | def get_scoped_name(self, name):
    method __iter__ (line 55) | def __iter__(self):
  class BasePlotterTasks (line 61) | class BasePlotterTasks(object):
    method __init__ (line 62) | def __init__(self, tasks) -> None:
    method __iter__ (line 65) | def __iter__(self):
  class BasePlotter (line 70) | class BasePlotter(object):
    method __init__ (line 77) | def __init__(self, task: BasePlotterTask) -> None:
    method task_primitives (line 82) | def task_primitives(self):
    method create (line 85) | def create(self, task: BasePlotterTask) -> None:
    method update (line 91) | def update(self) -> None:
    method _update_impl (line 95) | def _update_impl(self, task_list: List[BasePlotterTask]) -> None:
    method _create_impl (line 98) | def _create_impl(self, task_list: List[BasePlotterTask]) -> None:

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/visualization/plt_plotter.py
  class Matplotlib2DPlotter (line 44) | class Matplotlib2DPlotter(BasePlotter):
    method __init__ (line 53) | def __init__(self, task: "BasePlotterTask") -> None:
    method ax (line 73) | def ax(self):
    method fig (line 77) | def fig(self):
    method show (line 80) | def show(self):
    method _min (line 83) | def _min(self, x, y):
    method _max (line 90) | def _max(self, x, y):
    method _init_lim (line 97) | def _init_lim(self):
    method _update_lim (line 103) | def _update_lim(self, xs, ys):
    method _set_lim (line 109) | def _set_lim(self):
    method _lines_extract_xy_impl (line 121) | def _lines_extract_xy_impl(index, lines_task):
    method _trail_extract_xy_impl (line 125) | def _trail_extract_xy_impl(index, trail_task):
    method _lines_create_impl (line 128) | def _lines_create_impl(self, lines_task):
    method _lines_update_impl (line 140) | def _lines_update_impl(self, lines_task):
    method _dots_create_impl (line 149) | def _dots_create_impl(self, dots_task):
    method _dots_update_impl (line 161) | def _dots_update_impl(self, dots_task):
    method _trail_create_impl (line 167) | def _trail_create_impl(self, trail_task):
    method _trail_update_impl (line 180) | def _trail_update_impl(self, trail_task):
    method _create_impl (line 189) | def _create_impl(self, task_list):
    method _update_impl (line 194) | def _update_impl(self, task_list):
    method _set_aspect_equal_2d (line 199) | def _set_aspect_equal_2d(self, zero_centered=True):
    method _draw (line 221) | def _draw(self):
  class Matplotlib3DPlotter (line 229) | class Matplotlib3DPlotter(BasePlotter):
    method __init__ (line 238) | def __init__(self, task: "BasePlotterTask") -> None:
    method ax (line 257) | def ax(self):
    method fig (line 261) | def fig(self):
    method show (line 264) | def show(self):
    method _min (line 267) | def _min(self, x, y):
    method _max (line 274) | def _max(self, x, y):
    method _init_lim (line 281) | def _init_lim(self):
    method _update_lim (line 289) | def _update_lim(self, xs, ys, zs):
    method _set_lim (line 297) | def _set_lim(self):
    method _lines_extract_xyz_impl (line 312) | def _lines_extract_xyz_impl(index, lines_task):
    method _trail_extract_xyz_impl (line 316) | def _trail_extract_xyz_impl(index, trail_task):
    method _lines_create_impl (line 323) | def _lines_create_impl(self, lines_task):
    method _lines_update_impl (line 335) | def _lines_update_impl(self, lines_task):
    method _dots_create_impl (line 345) | def _dots_create_impl(self, dots_task):
    method _dots_update_impl (line 358) | def _dots_update_impl(self, dots_task):
    method _trail_create_impl (line 365) | def _trail_create_impl(self, trail_task):
    method _trail_update_impl (line 378) | def _trail_update_impl(self, trail_task):
    method _create_impl (line 388) | def _create_impl(self, task_list):
    method _update_impl (line 393) | def _update_impl(self, task_list):
    method _set_aspect_equal_3d (line 398) | def _set_aspect_equal_3d(self):
    method _draw (line 419) | def _draw(self):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/visualization/simple_plotter_tasks.py
  class DrawXDLines (line 38) | class DrawXDLines(BasePlotterTask):
    method __init__ (line 45) | def __init__(
    method influence_lim (line 62) | def influence_lim(self) -> bool:
    method raw_data (line 66) | def raw_data(self):
    method color (line 70) | def color(self):
    method line_width (line 74) | def line_width(self):
    method alpha (line 78) | def alpha(self):
    method dim (line 82) | def dim(self):
    method name (line 86) | def name(self):
    method update (line 89) | def update(self, lines):
    method __getitem__ (line 94) | def __getitem__(self, index):
    method __len__ (line 97) | def __len__(self):
    method __iter__ (line 100) | def __iter__(self):
  class DrawXDDots (line 104) | class DrawXDDots(BasePlotterTask):
    method __init__ (line 111) | def __init__(
    method update (line 127) | def update(self, dots):
    method __getitem__ (line 132) | def __getitem__(self, index):
    method __len__ (line 135) | def __len__(self):
    method __iter__ (line 138) | def __iter__(self):
    method influence_lim (line 142) | def influence_lim(self) -> bool:
    method raw_data (line 146) | def raw_data(self):
    method color (line 150) | def color(self):
    method marker_size (line 154) | def marker_size(self):
    method alpha (line 158) | def alpha(self):
    method dim (line 162) | def dim(self):
    method name (line 166) | def name(self):
  class DrawXDTrail (line 170) | class DrawXDTrail(DrawXDDots):
    method line_width (line 172) | def line_width(self):
    method name (line 176) | def name(self):
  class Draw2DLines (line 180) | class Draw2DLines(DrawXDLines):
    method dim (line 182) | def dim(self):
  class Draw3DLines (line 186) | class Draw3DLines(DrawXDLines):
    method dim (line 188) | def dim(self):
  class Draw2DDots (line 192) | class Draw2DDots(DrawXDDots):
    method dim (line 194) | def dim(self):
  class Draw3DDots (line 198) | class Draw3DDots(DrawXDDots):
    method dim (line 200) | def dim(self):
  class Draw2DTrail (line 204) | class Draw2DTrail(DrawXDTrail):
    method dim (line 206) | def dim(self):
  class Draw3DTrail (line 210) | class Draw3DTrail(DrawXDTrail):
    method dim (line 212) | def dim(self):

FILE: timechamber/tasks/ase_humanoid_base/poselib/poselib/visualization/skeleton_plotter_tasks.py
  class Draw3DSkeletonState (line 40) | class Draw3DSkeletonState(BasePlotterTask):
    method __init__ (line 44) | def __init__(
    method name (line 62) | def name(self):
    method update (line 65) | def update(self, skeleton_state) -> None:
    method _get_lines_and_dots (line 69) | def _get_lines_and_dots(skeleton_state):
    method _update (line 86) | def _update(self, lines, dots) -> None:
    method __iter__ (line 90) | def __iter__(self):
  class Draw3DSkeletonMotion (line 95) | class Draw3DSkeletonMotion(BasePlotterTask):
    method __init__ (line 96) | def __init__(
    method name (line 155) | def name(self):
    method update (line 158) | def update(self, frame_index=None, reset_trail=False, skeleton_motion=...
    method _get_vel_and_avel (line 182) | def _get_vel_and_avel(skeleton_motion):
    method _update (line 193) | def _update(self, vel_lines, avel_lines) -> None:
    method __iter__ (line 197) | def __iter__(self):
  class Draw3DSkeletonMotions (line 204) | class Draw3DSkeletonMotions(BasePlotterTask):
    method __init__ (line 205) | def __init__(self, skeleton_motion_tasks) -> None:
    method name (line 209) | def name(self):
    method update (line 212) | def update(self, frame_index) -> None:
    method __iter__ (line 215) | def __iter__(self):

FILE: timechamber/tasks/ase_humanoid_base/poselib/retarget_motion.py
  function project_joints (line 52) | def project_joints(motion):
  function main (line 206) | def main():

FILE: timechamber/tasks/base/ma_vec_task.py
  class MA_VecTask (line 50) | class MA_VecTask(Env):
    method __init__ (line 52) | def __init__(self, config, rl_device, sim_device, graphics_device_id, ...
    method set_viewer (line 103) | def set_viewer(self):
    method allocate_buffers (line 132) | def allocate_buffers(self):
    method set_sim_params_up_axis (line 157) | def set_sim_params_up_axis(self, sim_params: gymapi.SimParams, axis: s...
    method create_sim (line 174) | def create_sim(self, compute_device: int, graphics_device: int, physic...
    method get_state (line 192) | def get_state(self):
    method pre_physics_step (line 197) | def pre_physics_step(self, actions: torch.Tensor):
    method post_physics_step (line 205) | def post_physics_step(self):
    method step (line 208) | def step(self, actions: torch.Tensor) -> Tuple[Dict[str, torch.Tensor]...
    method zero_actions (line 250) | def zero_actions(self) -> torch.Tensor:
    method reset (line 261) | def reset(self, env_ids=None) -> torch.Tensor:
    method _reset_envs (line 275) | def _reset_envs(self, env_ids):
    method reset_done (line 282) | def reset_done(self):
    method render (line 298) | def render(self):
    method __parse_sim_params (line 328) | def __parse_sim_params(self, physics_engine: str, config_sim: Dict[str...
    method get_actor_params_info (line 382) | def get_actor_params_info(self, dr_params: Dict[str, Any], env):
    method apply_randomizations (line 424) | def apply_randomizations(self, dr_params):

FILE: timechamber/tasks/base/vec_task.py
  function _create_sim_once (line 50) | def _create_sim_once(gym, *args, **kwargs):
  class Env (line 59) | class Env(ABC):
    method __init__ (line 60) | def __init__(self, config: Dict[str, Any], rl_device: str, sim_device:...
    method allocate_buffers (line 110) | def allocate_buffers(self):
    method step (line 114) | def step(self, actions: torch.Tensor) -> Tuple[Dict[str, torch.Tensor]...
    method reset (line 125) | def reset(self)-> Dict[str, torch.Tensor]:
    method reset_idx (line 132) | def reset_idx(self, env_ids: torch.Tensor):
    method observation_space (line 139) | def observation_space(self) -> gym.Space:
    method action_space (line 144) | def action_space(self) -> gym.Space:
    method num_envs (line 149) | def num_envs(self) -> int:
    method num_acts (line 154) | def num_acts(self) -> int:
    method num_obs (line 159) | def num_obs(self) -> int:
  class VecTask (line 164) | class VecTask(Env):
    method __init__ (line 168) | def __init__(self, config, rl_device, sim_device, graphics_device_id, ...
    method set_viewer (line 224) | def set_viewer(self):
    method allocate_buffers (line 253) | def allocate_buffers(self):
    method create_sim (line 278) | def create_sim(self, compute_device: int, graphics_device: int, physic...
    method get_state (line 296) | def get_state(self):
    method pre_physics_step (line 301) | def pre_physics_step(self, actions: torch.Tensor):
    method post_physics_step (line 309) | def post_physics_step(self):
    method step (line 312) | def step(self, actions: torch.Tensor) -> Tuple[Dict[str, torch.Tensor]...
    method zero_actions (line 360) | def zero_actions(self) -> torch.Tensor:
    method reset_idx (line 370) | def reset_idx(self, env_idx):
    method reset (line 376) | def reset(self):
    method reset_done (line 390) | def reset_done(self):
    method render (line 407) | def render(self, mode="rgb_array"):
    method __parse_sim_params (line 441) | def __parse_sim_params(self, physics_engine: str, config_sim: Dict[str...
    method get_actor_params_info (line 495) | def get_actor_params_info(self, dr_params: Dict[str, Any], env):
    method apply_randomizations (line 537) | def apply_randomizations(self, dr_params):

FILE: timechamber/tasks/ma_ant_battle.py
  class MA_Ant_Battle (line 12) | class MA_Ant_Battle(MA_VecTask):
    method __init__ (line 14) | def __init__(self, cfg, sim_device, rl_device, graphics_device_id, hea...
    method allocate_buffers (line 102) | def allocate_buffers(self):
    method create_sim (line 122) | def create_sim(self):
    method _add_circle_borderline (line 142) | def _add_circle_borderline(self, env, radius):
    method _create_ground_plane (line 147) | def _create_ground_plane(self):
    method _create_envs (line 154) | def _create_envs(self, num_envs, spacing, num_per_row):
    method compute_reward (line 244) | def compute_reward(self, actions):
    method compute_observations (line 268) | def compute_observations(self):
    method reset_idx (line 287) | def reset_idx(self, env_ids):
    method pre_physics_step (line 328) | def pre_physics_step(self, actions):
    method post_physics_step (line 349) | def post_physics_step(self):
    method get_number_of_agents (line 368) | def get_number_of_agents(self):
    method zero_actions (line 372) | def zero_actions(self) -> torch.Tensor:
    method clear_count (line 383) | def clear_count(self):
  function expand_env_ids (line 394) | def expand_env_ids(env_ids, n_agents):
  function compute_ant_reward (line 405) | def compute_ant_reward(
  function compute_ant_observations (line 465) | def compute_ant_observations(
  function randomize_rotation (line 501) | def randomize_rotation(rand0, rand1, x_unit_tensor, y_unit_tensor):

FILE: timechamber/tasks/ma_ant_sumo.py
  class MA_Ant_Sumo (line 19) | class MA_Ant_Sumo(MA_VecTask):
    method __init__ (line 21) | def __init__(self, cfg, sim_device, rl_device, graphics_device_id, hea...
    method allocate_buffers (line 113) | def allocate_buffers(self):
    method create_sim (line 130) | def create_sim(self):
    method _add_circle_borderline (line 142) | def _add_circle_borderline(self, env):
    method _create_ground_plane (line 155) | def _create_ground_plane(self):
    method _create_envs (line 162) | def _create_envs(self, num_envs, spacing, num_per_row):
    method compute_reward (line 271) | def compute_reward(self, actions):
    method compute_observations (line 298) | def compute_observations(self):
    method reset_idx (line 326) | def reset_idx(self, env_ids):
    method pre_physics_step (line 374) | def pre_physics_step(self, actions):
    method post_physics_step (line 388) | def post_physics_step(self):
    method get_number_of_agents (line 396) | def get_number_of_agents(self):
    method zero_actions (line 400) | def zero_actions(self) -> torch.Tensor:
    method clear_count (line 411) | def clear_count(self):
  function expand_env_ids (line 423) | def expand_env_ids(env_ids, n_agents):
  function compute_move_reward (line 433) | def compute_move_reward(
  function compute_ant_reward (line 449) | def compute_ant_reward(
  function compute_ant_observations (line 508) | def compute_ant_observations(
  function randomize_rotation (line 529) | def randomize_rotation(rand0, rand1, x_unit_tensor, y_unit_tensor):

FILE: timechamber/tasks/ma_humanoid_strike.py
  class HumanoidStrike (line 41) | class HumanoidStrike(humanoid_amp_task.HumanoidAMPTask):
    method __init__ (line 42) | def __init__(self, cfg, sim_params, physics_engine, device_type, devic...
    method get_task_obs_size (line 80) | def get_task_obs_size(self):
    method _create_envs (line 86) | def _create_envs(self, num_envs, spacing, num_per_row):
    method _build_env (line 91) | def _build_env(self, env_id, env_ptr, humanoid_asset, humanoid_asset_op):
    method _build_body_ids_tensor (line 95) | def _build_body_ids_tensor(self, env_ptr, actor_handle, body_names):
    method _reset_actors (line 108) | def _reset_actors(self, env_ids):
    method _reset_env_tensors (line 141) | def _reset_env_tensors(self, env_ids):
    method pre_physics_step (line 147) | def pre_physics_step(self, actions):
    method post_physics_step (line 154) | def post_physics_step(self):
    method _compute_observations (line 159) | def _compute_observations(self):
    method _compute_task_obs (line 170) | def _compute_task_obs(self):
    method _compute_reward (line 197) | def _compute_reward(self, actions):
    method _compute_reset (line 233) | def _compute_reset(self):
  function compute_strike_observations (line 255) | def compute_strike_observations(root_states, root_states_op, body_pos, b...
  function compute_strike_reward (line 343) | def compute_strike_reward(root_states, root_states_op, body_pos, body_an...
  function compute_humanoid_reset (line 522) | def compute_humanoid_reset(reset_buf, progress_buf, ego_to_op_damage, op...
  function expand_env_ids (line 593) | def expand_env_ids(env_ids, n_agents):

FILE: timechamber/train.py
  function launch_rlg_hydra (line 65) | def launch_rlg_hydra(cfg: DictConfig):

FILE: timechamber/utils/config.py
  function parse_sim_params (line 14) | def parse_sim_params(args, cfg):

FILE: timechamber/utils/gym_util.py
  function setup_gym_viewer (line 36) | def setup_gym_viewer(config):
  function initialize_gym (line 42) | def initialize_gym(config):
  function configure_gym (line 51) | def configure_gym(gym, config):
  function parse_states_from_reference_states (line 102) | def parse_states_from_reference_states(reference_states, progress):
  function parse_states_from_reference_states_with_motion_id (line 123) | def parse_states_from_reference_states_with_motion_id(precomputed_state,
  function parse_dof_state_with_motion_id (line 139) | def parse_dof_state_with_motion_id(precomputed_state, dof_state,
  function get_flatten_ids (line 152) | def get_flatten_ids(precomputed_state):
  function parse_states_from_reference_states_with_global_id (line 168) | def parse_states_from_reference_states_with_global_id(precomputed_state,
  function get_robot_states_from_torch_tensor (line 181) | def get_robot_states_from_torch_tensor(config, ts, global_quats, vels, a...
  function get_xyzoffset (line 235) | def get_xyzoffset(start_ts, end_ts, root_yaw_inv):

FILE: timechamber/utils/logger.py
  class _MyFormatter (line 19) | class _MyFormatter(logging.Formatter):
    method format (line 25) | def format(self, record):
  class GLOBAL_PATH (line 57) | class GLOBAL_PATH(object):
    method __init__ (line 59) | def __init__(self, path=None):
    method _set_path (line 64) | def _set_path(self, path):
    method _get_path (line 67) | def _get_path(self):
  function set_file_handler (line 74) | def set_file_handler(path=None, prefix='', time_str=''):
  function _get_path (line 107) | def _get_path():

FILE: timechamber/utils/motion_lib.py
  class Patch (line 46) | class Patch:
    method numpy (line 47) | def numpy(self):
  class DeviceCache (line 55) | class DeviceCache:
    method __init__ (line 56) | def __init__(self, obj, device):
    method __getattr__ (line 87) | def __getattr__(self, string):
  class MotionLib (line 92) | class MotionLib():
    method __init__ (line 93) | def __init__(self, motion_file, dof_body_ids, dof_offsets,
    method num_motions (line 119) | def num_motions(self):
    method get_total_length (line 122) | def get_total_length(self):
    method get_motion (line 125) | def get_motion(self, motion_id):
    method sample_motions (line 128) | def sample_motions(self, n):
    method sample_time (line 136) | def sample_time(self, motion_ids, truncate_time=None):
    method get_motion_length (line 148) | def get_motion_length(self, motion_ids):
    method get_motion_state (line 151) | def get_motion_state(self, motion_ids, motion_times):
    method _load_motions (line 202) | def _load_motions(self, motion_file):
    method _fetch_motion_files (line 266) | def _fetch_motion_files(self, motion_file):
    method _calc_frame_blend (line 291) | def _calc_frame_blend(self, time, len, num_frames, dt):
    method _get_num_bodies (line 302) | def _get_num_bodies(self):
    method _compute_motion_dof_vels (line 307) | def _compute_motion_dof_vels(self, motion):
    method _local_rotation_to_dof (line 324) | def _local_rotation_to_dof(self, local_rot):
    method _local_rotation_to_dof_vel (line 354) | def _local_rotation_to_dof_vel(self, local_rot0, local_rot1, dt):

FILE: timechamber/utils/reformat.py
  function omegaconf_to_dict (line 32) | def omegaconf_to_dict(d: DictConfig)->Dict:
  function print_dict (line 42) | def print_dict(val, nesting: int = -4, start: bool = True):

FILE: timechamber/utils/rlgames_utils.py
  function get_rlgames_env_creator (line 45) | def get_rlgames_env_creator(
  class RLGPUAlgoObserver (line 122) | class RLGPUAlgoObserver(AlgoObserver):
    method __init__ (line 125) | def __init__(self):
    method after_init (line 128) | def after_init(self, algo):
    method process_infos (line 135) | def process_infos(self, infos, done_indices):
    method after_clear_stats (line 148) | def after_clear_stats(self):
    method after_print_stats (line 151) | def after_print_stats(self, frame, epoch_num, total_time):
  class RLGPUEnv (line 178) | class RLGPUEnv(vecenv.IVecEnv):
    method __init__ (line 179) | def __init__(self, config_name, num_actors, **kwargs):
    method step (line 189) | def step(self, action):
    method reset (line 198) | def reset(self, env_ids=None):
    method get_number_of_agents (line 204) | def get_number_of_agents(self):
    method get_env_info (line 207) | def get_env_info(self):

FILE: timechamber/utils/torch_jit_utils.py
  function compute_heading_and_up (line 35) | def compute_heading_and_up(
  function compute_rot (line 53) | def compute_rot(torso_quat, velocity, ang_velocity, targets, torso_posit...
  function quat_axis (line 67) | def quat_axis(q, axis=0):
  function scale_transform (line 80) | def scale_transform(x: torch.Tensor, lower: torch.Tensor, upper: torch.T...
  function unscale_transform (line 101) | def unscale_transform(x: torch.Tensor, lower: torch.Tensor, upper: torch...
  function saturate (line 121) | def saturate(x: torch.Tensor, lower: torch.Tensor, upper: torch.Tensor) ...
  function quat_diff_rad (line 142) | def quat_diff_rad(a: torch.Tensor, b: torch.Tensor) -> torch.Tensor:
  function local_to_world_space (line 164) | def local_to_world_space(pos_offset_local: torch.Tensor, pose_global: to...
  function normalise_quat_in_pose (line 185) | def normalise_quat_in_pose(pose):
  function my_quat_rotate (line 199) | def my_quat_rotate(q, v):
  function quat_to_angle_axis (line 211) | def quat_to_angle_axis(q):
  function angle_axis_to_exp_map (line 234) | def angle_axis_to_exp_map(angle, axis):
  function quat_to_exp_map (line 242) | def quat_to_exp_map(q):
  function quat_to_tan_norm (line 251) | def quat_to_tan_norm(q):
  function euler_xyz_to_exp_map (line 266) | def euler_xyz_to_exp_map(roll, pitch, yaw):
  function exp_map_to_angle_axis (line 273) | def exp_map_to_angle_axis(exp_map):
  function exp_map_to_quat (line 292) | def exp_map_to_quat(exp_map):
  function slerp (line 298) | def slerp(q0, q1, t):
  function calc_heading (line 333) | def calc_heading(q):
  function calc_heading_quat (line 346) | def calc_heading_quat(q):
  function calc_heading_quat_inv (line 359) | def calc_heading_quat_inv(q):

FILE: timechamber/utils/torch_utils.py
  function quat_to_angle_axis (line 35) | def quat_to_angle_axis(q):
  function angle_axis_to_exp_map (line 58) | def angle_axis_to_exp_map(angle, axis):
  function quat_to_exp_map (line 66) | def quat_to_exp_map(q):
  function quat_to_tan_norm (line 75) | def quat_to_tan_norm(q):
  function euler_xyz_to_exp_map (line 90) | def euler_xyz_to_exp_map(roll, pitch, yaw):
  function exp_map_to_angle_axis (line 97) | def exp_map_to_angle_axis(exp_map):
  function exp_map_to_quat (line 116) | def exp_map_to_quat(exp_map):
  function slerp (line 122) | def slerp(q0, q1, t):
  function calc_heading (line 146) | def calc_heading(q):
  function calc_heading_quat (line 159) | def calc_heading_quat(q):
  function calc_heading_quat_inv (line 172) | def calc_heading_quat_inv(q):

FILE: timechamber/utils/utils.py
  function set_np_formatting (line 40) | def set_np_formatting():
  function set_seed (line 47) | def set_seed(seed, torch_deterministic=False, rank=0):
  function load_check (line 77) | def load_check(checkpoint, normalize_input: bool, normalize_value: bool):
  function safe_filesystem_op (line 93) | def safe_filesystem_op(func, *args, **kwargs):
  function safe_load (line 110) | def safe_load(filename, device=None):
  function load_checkpoint (line 116) | def load_checkpoint(filename, device=None):
  function print_actor_info (line 121) | def print_actor_info(gym, env, actor_handle):
  function print_asset_info (line 182) | def print_asset_info(asset, name, gym):

FILE: timechamber/utils/vec_task.py
  class VecTask (line 17) | class VecTask():
    method __init__ (line 18) | def __init__(self, task, rl_device, clip_observations=5.0, clip_action...
    method step (line 37) | def step(self, actions):
    method reset (line 40) | def reset(self):
    method get_number_of_agents (line 43) | def get_number_of_agents(self):
    method observation_space (line 47) | def observation_space(self):
    method action_space (line 51) | def action_space(self):
    method num_envs (line 55) | def num_envs(self):
    method num_acts (line 59) | def num_acts(self):
    method num_obs (line 63) | def num_obs(self):
  class VecTaskCPU (line 68) | class VecTaskCPU(VecTask):
    method __init__ (line 69) | def __init__(self, task, rl_device, sync_frame_time=False, clip_observ...
    method step (line 73) | def step(self, actions):
    method reset (line 83) | def reset(self):
  class VecTaskGPU (line 93) | class VecTaskGPU(VecTask):
    method __init__ (line 94) | def __init__(self, task, rl_device, clip_observations=5.0, clip_action...
    method step (line 101) | def step(self, actions):
    method reset (line 110) | def reset(self):
  class VecTaskPython (line 121) | class VecTaskPython(VecTask):
    method get_state (line 123) | def get_state(self):
    method step (line 126) | def step(self, actions):
    method reset (line 133) | def reset(self):

FILE: timechamber/utils/vec_task_wrappers.py
  class VecTaskCPUWrapper (line 34) | class VecTaskCPUWrapper(VecTaskCPU):
    method __init__ (line 35) | def __init__(self, task, rl_device, sync_frame_time=False, clip_observ...
  class VecTaskGPUWrapper (line 39) | class VecTaskGPUWrapper(VecTaskGPU):
    method __init__ (line 40) | def __init__(self, task, rl_device, clip_observations=5.0, clip_action...
  class VecTaskPythonWrapper (line 45) | class VecTaskPythonWrapper(VecTaskPython):
    method __init__ (line 46) | def __init__(self, task, rl_device, clip_observations=5.0, clip_action...
    method reset (line 54) | def reset(self, env_ids=None):
    method amp_observation_space (line 59) | def amp_observation_space(self):
    method fetch_amp_obs_demo (line 62) | def fetch_amp_obs_demo(self, num_samples):