chore: move folder creation

lavague-ai · May 21, 2024 · c70ce5c · c70ce5c
1 parent 6cfca70
commit c70ce5c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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)