From 00ad4d4c7760e2a0ca3a7de3e9437cf75d982c58 Mon Sep 17 00:00:00 2001 From: recursix Date: Thu, 5 Dec 2024 08:51:26 -0500 Subject: [PATCH] fix: update demo_mode assignment in GenericAgentArgs class --- src/agentlab/agents/generic_agent/generic_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/agentlab/agents/generic_agent/generic_agent.py b/src/agentlab/agents/generic_agent/generic_agent.py index 98026dc1f..3db531bd2 100644 --- a/src/agentlab/agents/generic_agent/generic_agent.py +++ b/src/agentlab/agents/generic_agent/generic_agent.py @@ -43,7 +43,7 @@ def set_benchmark(self, benchmark: bgym.Benchmark, demo_mode): # verify if we can remove this if demo_mode: - self.action_set.demo_mode = "all_blue" + self.flags.action.action_set.demo_mode = "all_blue" def set_reproducibility_mode(self): self.chat_model_args.temperature = 0