removed the data saving loop, ran black, updated readme

Author: farhanishmam

README.md

@@ -63,6 +63,7 @@ AgentLab Features:
 | [Mind2Web-live](https://huggingface.co/datasets/iMeanAI/Mind2Web-Live) (soon) | - | - | None | - | - | live web | soon |
 | [MiniWoB](https://miniwob.farama.org/index.html) | [setup](https://github.com/ServiceNow/BrowserGym/blob/main/browsergym/miniwob/README.md) | 125 | Medium | 10 | no | self hosted (static files) | soon |
 | [OSWorld](https://os-world.github.io/) | [setup](https://github.com/ServiceNow/AgentLab/blob/main/src/agentlab/benchmarks/osworld.md) | 369 | None | - | - | self hosted  | soon |
+| [TimeWarp](https://timewarp-web.github.io/) | [setup](https://github.com/sparklabutah/timewarp) | 1386 | None | 30 | yes | self hosted | soon |
 
 
 ## 🛠️ Setup AgentLab

src/agentlab/agents/generic_agent/generic_agent_with_training.py

@@ -415,11 +415,11 @@ def run(self):
         try:
             logger.info(f"Running experiment {self.exp_name} in:\n  {self.exp_dir}")
             agent = self.agent_args.make_agent()
-            
+
             # Set training data directory for agents that support it
-            if hasattr(agent, 'set_training_data_dir'):
+            if hasattr(agent, "set_training_data_dir"):
                 agent.set_training_data_dir(Path(self.exp_dir))
-            
+
             if hasattr(agent, "set_task_name"):
                 agent.set_task_name(self.env_args.task_name)
 
@@ -441,11 +441,11 @@ def run(self):
 
             while not step_info.is_done:  # set a limit
                 logger.debug(f"Starting step {step_info.step}.")
-                
+
                 # Set current step for training data capture
-                if hasattr(agent, 'set_current_step'):
+                if hasattr(agent, "set_current_step"):
                     agent.set_current_step(step_info.step)
-                
+
                 action = step_info.from_action(agent)
                 logger.debug(f"Agent chose action:\n {action}")