Improved typing

TLSDC · web-flow · commit 9e85d8883fb3 · 2025-04-24T14:56:36.000-04:00
diff --git a/src/agentlab/ui_assistant.py b/src/agentlab/ui_assistant.py
@@ -7,7 +7,7 @@
 from agentlab.experiments.loop import EnvArgs, ExpArgs
 
 
-def make_exp_args(agent_args: AgentArgs, start_url="https://www.google.com"):
+def make_exp_args(agent_args: GenericAgentArgs, start_url="https://www.google.com"):
     try:
         agent_args.flags.action.action_set.demo_mode = "default"
     except AttributeError:
