Improved typing

Author: TLSDC

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: