.codecov.yml
.coveragerc
.dockerignore
.gitattributes
.gitignore
.pre-commit-config.yaml
.readthedocs.yml
CITATION.bib
Dockerfile
LICENSE
README.md
mypy.ini
pyproject.toml
setup.cfg
setup.py
.circleci/config.yml
.github/pull_request_template.md
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/workflows/publish-to-pypi.yml
benchmarking/README.md
benchmarking/compute_probability_of_improvement.py
benchmarking/run_all_benchmarks.sh
benchmarking/run_all_benchmarks_on_slurm.sh
benchmarking/run_benchmark_on_slurm.sh
benchmarking/sacred_output_to_csv.py
benchmarking/sacred_output_to_markdown_summary.py
benchmarking/util.py
ci/Xdummy-entrypoint.py
ci/build_and_activate_venv.ps1
ci/build_and_activate_venv.sh
ci/check_typeignore.py
ci/clean_notebooks.py
ci/xorg.conf
docs/.gitignore
docs/Makefile
docs/conf.py
docs/index.rst
docs/make.bat
docs/_static/css/custom.css
docs/_templates/autosummary/base.rst
docs/_templates/autosummary/class.rst
docs/_templates/autosummary/module.rst
docs/algorithms/airl.rst
docs/algorithms/bc.rst
docs/algorithms/dagger.rst
docs/algorithms/density.rst
docs/algorithms/gail.rst
docs/algorithms/mce_irl.rst
docs/algorithms/preference_comparisons.rst
docs/algorithms/sqil.rst
docs/development/developer.rst
docs/development/license.rst
docs/development/release-notes.rst
docs/development/contributing/code-of-conduct.rst
docs/development/contributing/index.rst
docs/development/contributing/ways-to-contribute.rst
docs/getting-started/cli.rst
docs/getting-started/first_steps.rst
docs/getting-started/installation.rst
docs/getting-started/what_is_imitation.rst
docs/main-concepts/benchmarks.md
docs/main-concepts/experts.rst
docs/main-concepts/reward_networks.rst
docs/main-concepts/trajectories.rst
docs/main-concepts/variable_horizon.rst
docs/tutorials/10_train_custom_env.ipynb
docs/tutorials/1_train_bc.ipynb
docs/tutorials/2_train_dagger.ipynb
docs/tutorials/3_train_gail.ipynb
docs/tutorials/4_train_airl.ipynb
docs/tutorials/5_train_preference_comparisons.ipynb
docs/tutorials/5a_train_preference_comparisons_with_cnn.ipynb
docs/tutorials/6_train_mce.ipynb
docs/tutorials/7_train_density.ipynb
docs/tutorials/8_train_sqil.ipynb
docs/tutorials/8a_train_sqil_sac.ipynb
docs/tutorials/9_compare_baselines.ipynb
examples/quickstart.py
examples/quickstart.sh
examples/train_dagger_atari_interactive_policy.py
experiments/.gitignore
experiments/README.md
experiments/bc_benchmark.sh
experiments/benchmark_and_table.sh
experiments/commands.py
experiments/common.sh
experiments/convert_traj.py
experiments/dagger_benchmark.sh
experiments/imit_benchmark.sh
experiments/imit_benchmark_config.csv
experiments/imit_table_cheetahs.csv
experiments/imit_table_mvp_seals_config.csv
experiments/rollouts_from_policies.sh
experiments/rollouts_from_policies_config.csv
experiments/transfer_learn_benchmark.sh
runners/build_push_image.sh
runners/launch_docker-dev.sh
src/imitation/__init__.py
src/imitation/py.typed
src/imitation.egg-info/PKG-INFO
src/imitation.egg-info/SOURCES.txt
src/imitation.egg-info/dependency_links.txt
src/imitation.egg-info/entry_points.txt
src/imitation.egg-info/requires.txt
src/imitation.egg-info/top_level.txt
src/imitation/algorithms/__init__.py
src/imitation/algorithms/base.py
src/imitation/algorithms/bc.py
src/imitation/algorithms/dagger.py
src/imitation/algorithms/density.py
src/imitation/algorithms/mce_irl.py
src/imitation/algorithms/preference_comparisons.py
src/imitation/algorithms/sqil.py
src/imitation/algorithms/adversarial/__init__.py
src/imitation/algorithms/adversarial/airl.py
src/imitation/algorithms/adversarial/common.py
src/imitation/algorithms/adversarial/gail.py
src/imitation/data/__init__.py
src/imitation/data/buffer.py
src/imitation/data/huggingface_utils.py
src/imitation/data/rollout.py
src/imitation/data/serialize.py
src/imitation/data/types.py
src/imitation/data/wrappers.py
src/imitation/policies/__init__.py
src/imitation/policies/base.py
src/imitation/policies/exploration_wrapper.py
src/imitation/policies/interactive.py
src/imitation/policies/replay_buffer_wrapper.py
src/imitation/policies/serialize.py
src/imitation/regularization/__init__.py
src/imitation/regularization/regularizers.py
src/imitation/regularization/updaters.py
src/imitation/rewards/__init__.py
src/imitation/rewards/reward_function.py
src/imitation/rewards/reward_nets.py
src/imitation/rewards/reward_wrapper.py
src/imitation/rewards/serialize.py
src/imitation/scripts/__init__.py
src/imitation/scripts/analyze.py
src/imitation/scripts/convert_trajs.py
src/imitation/scripts/eval_policy.py
src/imitation/scripts/parallel.py
src/imitation/scripts/train_adversarial.py
src/imitation/scripts/train_imitation.py
src/imitation/scripts/train_preference_comparisons.py
src/imitation/scripts/train_rl.py
src/imitation/scripts/tuning.py
src/imitation/scripts/config/__init__.py
src/imitation/scripts/config/analyze.py
src/imitation/scripts/config/eval_policy.py
src/imitation/scripts/config/parallel.py
src/imitation/scripts/config/train_adversarial.py
src/imitation/scripts/config/train_imitation.py
src/imitation/scripts/config/train_preference_comparisons.py
src/imitation/scripts/config/train_rl.py
src/imitation/scripts/config/tuning.py
src/imitation/scripts/config/tuned_hps/airl_seals_ant_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/airl_seals_half_cheetah_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/airl_seals_hopper_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/airl_seals_swimmer_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/airl_seals_walker_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/bc_seals_ant_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/bc_seals_half_cheetah_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/bc_seals_hopper_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/bc_seals_swimmer_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/bc_seals_walker_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/dagger_seals_ant_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/dagger_seals_half_cheetah_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/dagger_seals_hopper_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/dagger_seals_swimmer_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/dagger_seals_walker_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/fast_dagger_seals_cartpole.json
src/imitation/scripts/config/tuned_hps/gail_seals_ant_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/gail_seals_half_cheetah_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/gail_seals_hopper_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/gail_seals_swimmer_best_hp_eval.json
src/imitation/scripts/config/tuned_hps/gail_seals_walker_best_hp_eval.json
src/imitation/scripts/ingredients/__init__.py
src/imitation/scripts/ingredients/bc.py
src/imitation/scripts/ingredients/demonstrations.py
src/imitation/scripts/ingredients/environment.py
src/imitation/scripts/ingredients/expert.py
src/imitation/scripts/ingredients/logging.py
src/imitation/scripts/ingredients/policy.py
src/imitation/scripts/ingredients/policy_evaluation.py
src/imitation/scripts/ingredients/reward.py
src/imitation/scripts/ingredients/rl.py
src/imitation/scripts/ingredients/sqil.py
src/imitation/scripts/ingredients/wb.py
src/imitation/testing/__init__.py
src/imitation/testing/expert_trajectories.py
src/imitation/testing/hypothesis_strategies.py
src/imitation/testing/reward_improvement.py
src/imitation/testing/reward_nets.py
src/imitation/util/__init__.py
src/imitation/util/logger.py
src/imitation/util/networks.py
src/imitation/util/registry.py
src/imitation/util/sacred.py
src/imitation/util/sacred_file_parsing.py
src/imitation/util/util.py
src/imitation/util/video_wrapper.py
tests/conftest.py
tests/generate_test_data.sh
tests/test_benchmarking.py
tests/test_examples.py
tests/test_experiments.py
tests/test_regularization.py
tests/algorithms/__init__.py
tests/algorithms/conftest.py
tests/algorithms/test_adversarial.py
tests/algorithms/test_base.py
tests/algorithms/test_bc.py
tests/algorithms/test_dagger.py
tests/algorithms/test_density_baselines.py
tests/algorithms/test_mce_irl.py
tests/algorithms/test_preference_comparisons.py
tests/algorithms/test_sqil.py
tests/data/test_buffer.py
tests/data/test_huggingface_utils.py
tests/data/test_rollout.py
tests/data/test_types.py
tests/data/test_wrappers.py
tests/policies/test_exploration_wrapper.py
tests/policies/test_interactive.py
tests/policies/test_policies.py
tests/policies/test_replay_buffer_wrapper.py
tests/rewards/test_reward_fn.py
tests/rewards/test_reward_nets.py
tests/rewards/test_reward_wrapper.py
tests/scripts/test_scripts.py
tests/scripts/ingredients/test_rewards.py
tests/testdata/imit_benchmark_config.csv
tests/testdata/npz_format_rollout.npz
tests/testdata/pickle_format_rollout.pkl
tests/testdata/rollouts_from_policies_config.csv
tests/testdata/expert_models/cartpole_0/policies/final/model.zip
tests/testdata/expert_models/cartpole_0/rollouts/final.npz
tests/testdata/expert_models/pendulum_0/rollouts/final.npz
tests/util/test_logger.py
tests/util/test_networks.py
tests/util/test_registry.py
tests/util/test_sacred_file_parsing.py
tests/util/test_util.py
tests/util/test_wb_logger.py