From c70ce5ca8a547890c2b4b68001ba4809aa6d53b7 Mon Sep 17 00:00:00 2001 From: JoFrost Date: Tue, 21 May 2024 17:47:28 +0300 Subject: [PATCH] chore: move folder creation --- lavague-core/lavague/core/agents.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/lavague-core/lavague/core/agents.py b/lavague-core/lavague/core/agents.py index 928e1f76..80d62a84 100644 --- a/lavague-core/lavague/core/agents.py +++ b/lavague-core/lavague/core/agents.py @@ -37,9 +37,6 @@ def run(self, objective, display=True, log: bool = False): log_lines = [] - if log: - if os.path.isdir("./logs") == False: - os.mkdir("./logs") driver: WebDriver = self.driver.get_driver() action_engine: ActionEngine = self.action_engine world_model: WorldModel = self.world_model @@ -160,7 +157,9 @@ def run(self, objective, display=True, log: bool = False): if log: import csv - + + if os.path.isdir("./logs") == False: + os.mkdir("./logs") with open(f"./logs/{run_id}.csv", "w", newline="") as output_file: keys = log_lines[0].keys() dict_writer = csv.DictWriter(output_file, keys)