diff --git a/cooperative_cuisine/reinforcement_learning/train_single_agent.py b/cooperative_cuisine/reinforcement_learning/train_single_agent.py index 2801a23d804a1531f82f9313aef38933ba8e9607..633e6b7760e304daa54b51386f445a5751cb4ce6 100644 --- a/cooperative_cuisine/reinforcement_learning/train_single_agent.py +++ b/cooperative_cuisine/reinforcement_learning/train_single_agent.py @@ -33,7 +33,7 @@ def main(cfg: DictConfig): vec_env = additional_configs["vec_env"] number_envs_parallel = config["number_envs_parallel"] model_class = instantiate(cfg.model.model_type) - data_to_log=dict(config, **env_info) + data_to_log=dict(config, **env_info, **additional_configs) if vec_env: env = make_vec_env(lambda: EnvGymWrapper(cfg), n_envs=number_envs_parallel) else: