tests package

Subpackages

• tests.acceptance package
  • Submodules
  • tests.acceptance.test_calibration module
    • test_calibration_end_to_end()
  • tests.acceptance.test_config module
  • tests.acceptance.test_decision_transformer module
  • tests.acceptance.test_dt_train module
    • dt()
    • env()
    • environment_config()
    • models()
    • online_config()
    • run_config()
    • test_data_loader()
    • test_evaluate_dt_agent()
    • test_evaluate_dt_agent_with_trajectory_writer()
    • test_test_dt_agent()
    • trajectory_data_set()
    • transformer_model_config()
  • tests.acceptance.test_environment_registration module
    • test_register_envs()
  • tests.acceptance.test_multienvironments module
    • test_multienv_sampler()
  • tests.acceptance.test_offline_dataset module
  • tests.acceptance.test_ppo module
    • compare_state_dicts()
    • environment_config()
    • fc_agent()
    • large_transformer_model_config()
    • lstm_agent()
    • lstm_agents()
    • lstm_config()
    • online_config()
    • run_config()
    • test_empty_env()
    • test_empty_env_flat_one_hot()
    • test_fc_agent_model_checkpoint_saving_and_loading()
    • test_lstm_ppo_model_checkpoint_saving_and_loading()
    • test_lstm_ppo_model_load_saved_checkpoints()
    • test_ppo_agent_gym()
    • test_ppo_agent_minigrid()
    • test_ppo_agent_rollout_minibatches_minigrid()
    • test_ppo_traj_agent_rollout_and_learn_minibatches()
    • test_ppo_traj_agent_rollout_minibatches()
    • test_sample_from_agents()
    • test_traj_ppo_model_checkpoint_saving_and_loading()
    • transformer_agent()
    • transformer_model_config()
  • Module contents
• tests.end_end package
  • Submodules
  • tests.end_end.test_decision_transformer module
    • download_training_data()
    • offline_config()
    • test_clone_transformer()
    • test_decision_transformer()
    • test_decision_transformer_activation_fn()
    • test_decision_transformer_bespoke()
    • test_decision_transformer_gated_mlp()
    • test_decision_transformer_optimizer()
    • test_decision_transformer_preln()
    • test_decision_transformer_scheduler()
    • test_decision_transformer_state_embedding()
    • transformer_model_config()
  • tests.end_end.test_ppo module
    • test_ppo_runner()
    • test_ppo_runner_lstm_model()
    • test_ppo_runner_traj_model()
    • test_ppo_runner_traj_model_memory()
  • tests.end_end.test_ppo_dt_integration module
  • Module contents
• tests.unit package
  • Submodules
  • tests.unit.test_config module
    • test_environment_config()
    • test_lstm_model_config()
    • test_offline_train_config()
    • test_offline_train_config_raise_error_no_traj_path()
    • test_online_train_config()
    • test_run_config()
    • test_transformer_model_config()
  • tests.unit.test_constants module
    • test_twenty_idx_format_func()
  • tests.unit.test_environments module
    • test_make_env()
    • test_make_env_change_view_size()
    • test_make_env_flat_one_hot()
    • test_make_env_flat_one_hot_view_size_change()
    • test_make_env_fully_observed()
  • tests.unit.test_ppo_agent module
    • big_transformer_agent()
    • big_transformer_model_config()
    • environment_config()
    • fc_agent()
    • lstm_agent()
    • lstm_config()
    • online_config()
    • optimizer()
    • test_fc_agent_init()
    • test_fc_agent_layer_init()
    • test_fc_agent_learn()
    • test_fc_agent_make_optimizer()
    • test_fc_agent_rollout()
    • test_get_agent_fc_agent()
    • test_get_agent_lstm_agent()
    • test_get_agent_transformer_agent()
    • test_lstm_agent_init()
    • test_lstm_agent_rollout()
    • test_lstm_agent_rollout_learn()
    • test_ppo_agent_init()
    • test_ppo_scheduler_step()
    • test_transformer_agent_init()
    • test_transformer_agent_larger_context_init()
    • test_transformer_agent_larger_context_learn()
    • test_transformer_agent_larger_context_rollout()
    • test_transformer_agent_learn()
    • test_transformer_agent_rollout()
    • transformer_agent()
    • transformer_model_config()
  • tests.unit.test_ppo_memory module
    • memory()
    • online_config()
    • test_add_end_of_episode()
    • test_add_no_ending()
    • test_compute_advantages()
    • test_get_minibatch_indexes()
    • test_get_minibatch_indexes_recurrence()
    • test_get_minibatches_given_indices_contiguity_of_obs()
    • test_get_minibatches_given_indices_single_env_contiguity_of_obs()
    • test_get_minibatches_recurrance_memory_and_mask()
    • test_get_minibatches_recurrence()
    • test_get_minibatches_standard()
    • test_init()
    • test_minibatch_class()
  • tests.unit.test_ppo_utils module
    • test_get_obs_preprocessor_MiniGrid()
    • test_get_obs_preprocessor_MiniGrid_flat()
    • test_get_obs_preprocessor_MiniGrid_full_img_obs()
    • test_get_obs_preprocessor_MiniGrid_full_img_partial()
    • test_get_obs_preprocessor_MiniGrid_full_obs()
    • test_get_obs_preprocessor_probe1()
  • tests.unit.test_streamlit_environment module
    • decision_transformer_path()
    • test_get_env_and_dt()
  • tests.unit.test_trajectory_model module
  • tests.unit.test_utils module
    • environment_config()
    • online_config()
    • run_config()
    • test_trajectory_writer_lzma()
    • test_trajectory_writer_numpy()
    • test_trajectory_writer_torch()
  • Module contents

Module contents