From 2c8e8a58f6066318a903d79effc1b4494c38e9f6 Mon Sep 17 00:00:00 2001
From: Timothyxxx <384084775@qq.com>
Date: Thu, 5 Dec 2024 15:45:09 +0800
Subject: [PATCH] Fix minor bug caused by new logging feat in aguvis agent traj

---
 mm_agents/agent.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mm_agents/agent.py b/mm_agents/agent.py
index 9ab64e8..d947fb4 100644
--- a/mm_agents/agent.py
+++ b/mm_agents/agent.py
@@ -1077,7 +1077,10 @@ class PromptAgent:
 
             return actions
 
-    def reset(self):
+    def reset(self, _logger=None):
+        global logger
+        logger = _logger if _logger is not None else logging.getLogger("desktopenv.agent")
+
         self.thoughts = []
         self.actions = []
         self.observations = []