.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
README.md
pyproject.toml
.github/workflows/publish.yml
.github/workflows/ruff.yml
.github/workflows/tests.yml
docs/Makefile
docs/conf.py
docs/index.md
docs/_static/custom.css
docs/_static/testbed-icon.png
docs/_static/testbed-icon.svg
docs/_static/testbed-logo-dark.svg
docs/_static/testbed-logo-light.svg
docs/_templates/layout.html
docs/examples/constrained_designs.md
docs/examples/inference.md
docs/examples/location_finding.md
docs/examples/using_hooks.md
docs/guides/configuration.md
docs/guides/custom_methods.md
docs/guides/custom_tasks.md
docs/guides/distributions.md
docs/guides/inference.md
docs/guides/using_designers.md
docs/guides/using_tasks.md
docs/guides/wandb.md
testbed/__init__.py
testbed/core/__init__.py
testbed/core/constraints.py
testbed/core/design_distributions.py
testbed/core/distributions.py
testbed/core/hooks.py
testbed/core/load.py
testbed/core/logging.py
testbed/core/manager.py
testbed/core/rollout.py
testbed/core/setup.py
testbed/core/util.py
testbed/design_nets/__init__.py
testbed/design_nets/aline.py
testbed/design_nets/base.py
testbed/design_nets/deterministic.py
testbed/design_nets/encoder.py
testbed/design_nets/history.py
testbed/design_nets/modules.py
testbed/design_nets/rlboed.py
testbed/design_nets/stochastic.py
testbed/design_nets/variational_posterior.py
testbed/design_objectives/__init__.py
testbed/design_objectives/barber_agakov.py
testbed/design_objectives/base_objective.py
testbed/design_objectives/contrastive.py
testbed/design_objectives/log_loss.py
testbed/design_objectives/loss.py
testbed/design_objectives/reinforce.py
testbed/design_objectives/snmc.py
testbed/design_objectives/spce.py
testbed/designers/__init__.py
testbed/designers/aline.py
testbed/designers/amortized.py
testbed/designers/base.py
testbed/designers/non_adaptive.py
testbed/designers/sequential.py
testbed/envs/__init__.py
testbed/envs/bed_env.py
testbed/envs/history.py
testbed/envs/rewards.py
testbed/eval/__init__.py
testbed/eval/base.py
testbed/eval/eval_loss.py
testbed/eval/eval_policy.py
testbed/eval/eval_random.py
testbed/eval/eval_static.py
testbed/inference/__init__.py
testbed/inference/importance_sampler.py
testbed/inference/log_posterior.py
testbed/inference/pyro_sampler.py
testbed/methods/__init__.py
testbed/methods/actionbed.py
testbed/methods/aline.py
testbed/methods/dad.py
testbed/methods/posthoc_action.py
testbed/methods/rlboed.py
testbed/methods/static.py
testbed/methods/var_post.py
testbed/recipes/experiments/cart_pole/actionbed.yaml
testbed/recipes/experiments/cart_pole/aline.yaml
testbed/recipes/experiments/cart_pole/dad.yaml
testbed/recipes/experiments/cart_pole/joint_var_post.yaml
testbed/recipes/experiments/cart_pole/posthoc_action.yaml
testbed/recipes/experiments/cart_pole/rlboed.yaml
testbed/recipes/experiments/cart_pole/static.yaml
testbed/recipes/experiments/ces/actionbed.yaml
testbed/recipes/experiments/ces/aline.yaml
testbed/recipes/experiments/ces/dad.yaml
testbed/recipes/experiments/ces/joint_var_post.yaml
testbed/recipes/experiments/ces/posthoc_action.yaml
testbed/recipes/experiments/ces/rlboed.yaml
testbed/recipes/experiments/ces/static.yaml
testbed/recipes/experiments/double_link/actionbed.yaml
testbed/recipes/experiments/double_link/aline.yaml
testbed/recipes/experiments/double_link/dad.yaml
testbed/recipes/experiments/double_link/joint_var_post.yaml
testbed/recipes/experiments/double_link/posthoc_action.yaml
testbed/recipes/experiments/double_link/rlboed.yaml
testbed/recipes/experiments/double_link/static.yaml
testbed/recipes/experiments/location_finding/posthoc_action.yaml
testbed/recipes/experiments/location_finding/rlboed.yaml
testbed/recipes/experiments/location_finding/var_post.yaml
testbed/recipes/experiments/location_finding_2src/actionbed.yaml
testbed/recipes/experiments/location_finding_2src/aline.yaml
testbed/recipes/experiments/location_finding_2src/dad.yaml
testbed/recipes/experiments/location_finding_2src/joint_var_post.yaml
testbed/recipes/experiments/location_finding_2src/posthoc_action.yaml
testbed/recipes/experiments/location_finding_2src/rlboed.yaml
testbed/recipes/experiments/location_finding_2src/static.yaml
testbed/recipes/experiments/michaelis_menten/actionbed.yaml
testbed/recipes/experiments/michaelis_menten/aline.yaml
testbed/recipes/experiments/michaelis_menten/dad.yaml
testbed/recipes/experiments/michaelis_menten/joint_var_post.yaml
testbed/recipes/experiments/michaelis_menten/posthoc_action.yaml
testbed/recipes/experiments/michaelis_menten/rlboed.yaml
testbed/recipes/experiments/michaelis_menten/static.yaml
testbed/recipes/experiments/pharmacokinetic/actionbed.yaml
testbed/recipes/experiments/pharmacokinetic/aline.yaml
testbed/recipes/experiments/pharmacokinetic/dad.yaml
testbed/recipes/experiments/pharmacokinetic/joint_var_post.yaml
testbed/recipes/experiments/pharmacokinetic/posthoc_action.yaml
testbed/recipes/experiments/pharmacokinetic/rlboed.yaml
testbed/recipes/experiments/pharmacokinetic/static.yaml
testbed/recipes/experiments/prey_population/rlboed.yaml
testbed/recipes/experiments/stochastic_pendulum/actionbed.yaml
testbed/recipes/experiments/stochastic_pendulum/aline.yaml
testbed/recipes/experiments/stochastic_pendulum/dad.yaml
testbed/recipes/experiments/stochastic_pendulum/joint_var_post.yaml
testbed/recipes/experiments/stochastic_pendulum/posthoc_action.yaml
testbed/recipes/experiments/stochastic_pendulum/rlboed.yaml
testbed/recipes/experiments/stochastic_pendulum/static.yaml
testbed/recipes/experiments/temporal_discounting/actionbed.yaml
testbed/recipes/experiments/temporal_discounting/aline.yaml
testbed/recipes/experiments/temporal_discounting/dad.yaml
testbed/recipes/experiments/temporal_discounting/joint_var_post.yaml
testbed/recipes/experiments/temporal_discounting/posthoc_action.yaml
testbed/recipes/experiments/temporal_discounting/rlboed.yaml
testbed/recipes/experiments/temporal_discounting/static.yaml
testbed/recipes/experiments/temporal_discounting/var_post.yaml
testbed/recipes/final_eval/default.yaml
testbed/recipes/hydra_run/default.yaml
testbed/recipes/method/actionbed.yaml
testbed/recipes/method/aline.yaml
testbed/recipes/method/dad.yaml
testbed/recipes/method/posthoc_action.yaml
testbed/recipes/method/rlboed.yaml
testbed/recipes/method/static.yaml
testbed/recipes/method/var_post.yaml
testbed/recipes/net/attentive_no_time_policy.yaml
testbed/recipes/net/attentive_policy.yaml
testbed/recipes/net/deepset_large_policy.yaml
testbed/recipes/net/deepset_large_time_policy.yaml
testbed/recipes/net/deepset_policy.yaml
testbed/recipes/net/deepset_small_time_policy.yaml
testbed/recipes/net/gated_deepset_policy.yaml
testbed/recipes/net/mixture_posterior.yaml
testbed/recipes/system/default.yaml
testbed/recipes/task/cart_pole.yaml
testbed/recipes/task/ces.yaml
testbed/recipes/task/double_link.yaml
testbed/recipes/task/location_finding.yaml
testbed/recipes/task/location_finding_2src.yaml
testbed/recipes/task/michaelis_menten.yaml
testbed/recipes/task/pharmacokinetic.yaml
testbed/recipes/task/prey_population.yaml
testbed/recipes/task/stochastic_pendulum.yaml
testbed/recipes/task/temporal_discounting.yaml
testbed/recipes/wandb/default.yaml
testbed/tasks/__init__.py
testbed/tasks/base.py
testbed/tasks/cart_pole.py
testbed/tasks/ces.py
testbed/tasks/double_link.py
testbed/tasks/gravimetry.py
testbed/tasks/linear_regression.py
testbed/tasks/location_finding.py
testbed/tasks/michaelis_menten.py
testbed/tasks/pharmacokinetic.py
testbed/tasks/prey_population.py
testbed/tasks/stochastic_pendulum.py
testbed/tasks/temporal_discounting.py
tests/test_actionbed_location_finding.py
tests/test_actionbed_reinforce.py
tests/test_aline_location_finding.py
tests/test_amortized_designer.py
tests/test_cart_pole.py
tests/test_constant_elasticity_of_substitution.py
tests/test_contrastive_scoring.py
tests/test_dad_location_finding.py
tests/test_design_objectives.py
tests/test_design_spaces.py
tests/test_designer_step_contract.py
tests/test_distributions.py
tests/test_double_link.py
tests/test_envs.py
tests/test_eval_loss.py
tests/test_gravimetry.py
tests/test_history_convention.py
tests/test_inference.py
tests/test_linear_regression.py
tests/test_load_designer.py
tests/test_location_finding.py
tests/test_log_loss.py
tests/test_logging.py
tests/test_losses.py
tests/test_michaelis_menten.py
tests/test_myopic_barber_agakov_history.py
tests/test_myopic_contrastive_history.py
tests/test_optional_hydra.py
tests/test_package_imports.py
tests/test_pharmacokinetic.py
tests/test_policy_contracts.py
tests/test_posthoc_action.py
tests/test_prey_population.py
tests/test_prior_contract.py
tests/test_random_designer.py
tests/test_recipes.py
tests/test_reinforce.py
tests/test_rlboed_location_finding.py
tests/test_rollout_hooks.py
tests/test_sequential_designer.py
tests/test_static_designer.py
tests/test_static_method.py
tests/test_stochastic_pendulum.py
tests/test_stochastic_policies.py
tests/test_task_history.py
tests/test_task_outcome_gradients.py
tests/test_temporal_discounting.py
tests/test_util.py
tests/test_var_post.py
torch_testbed.egg-info/PKG-INFO
torch_testbed.egg-info/SOURCES.txt
torch_testbed.egg-info/dependency_links.txt
torch_testbed.egg-info/requires.txt
torch_testbed.egg-info/scm_file_list.json
torch_testbed.egg-info/scm_version.json
torch_testbed.egg-info/top_level.txt