diff --git a/ding/entry/serial_entry_mbrl.py b/ding/entry/serial_entry_mbrl.py index 426e4ca667..edb97e0c62 100644 --- a/ding/entry/serial_entry_mbrl.py +++ b/ding/entry/serial_entry_mbrl.py @@ -78,9 +78,7 @@ def mbrl_entry_setup( cfg.policy.other.commander, learner, collector, evaluator, env_buffer, policy.command_mode ) - return ( - cfg, policy, world_model, env_buffer, learner, collector, collector_env, evaluator, commander, tb_logger - ) + return (cfg, policy, world_model, env_buffer, learner, collector, collector_env, evaluator, commander, tb_logger) def create_img_buffer(