it was comented out for cirular import, but seems like there no cirular import anymore... I think

Author: recursix

src/agentlab/experiments/loop.py

@@ -25,7 +25,7 @@
 from PIL import Image
 from tqdm import tqdm
 
-# from agentlab.agents.tapeagent import TapeAgent, save_tape
+from agentlab.agents.tapeagent import TapeAgent, save_tape
 
 logger = logging.getLogger(__name__)
 
@@ -236,10 +236,6 @@ def make_stats(self):
             stats = {}
         stats.update(self.agent_info.pop("stats", {}))
 
-        # messages = self.agent_info.get("chat_messages", None)
-        # if messages is not None:
-        #     stats["n_token_agent_messages"] = count_messages_token(messages)
-
         t = self.profiling
         stats["step_elapsed"] = t.env_stop - t.env_start
         stats["agent_elapsed"] = t.agent_stop - t.agent_start
@@ -477,9 +473,9 @@ def run(self):
                     err_msg = f"Exception uncaught by agent or environment in task {self.env_args.task_name}.\n{type(e).__name__}:\n{e}"
                 logger.info("Saving experiment info.")
                 self.save_summary_info(episode_info, Path(self.exp_dir), err_msg, stack_trace)
-                # if isinstance(agent, TapeAgent):
-                #     task = getattr(env, "task", {})
-                #     save_tape(self.exp_dir, episode_info, task, agent.final_tape)
+                if isinstance(agent, TapeAgent):
+                    task = getattr(env, "task", {})
+                    save_tape(self.exp_dir, episode_info, task, agent.final_tape)
             except Exception as e:
                 logger.exception(f"Error while saving experiment info: {e}")
             try: