gaia2 typing fix

cemde · cemde · commit 211c087e2a8b · 2026-02-12T15:25:52.000+01:00
diff --git a/maseval/benchmark/gaia2/gaia2.py b/maseval/benchmark/gaia2/gaia2.py
@@ -1217,7 +1217,7 @@ def setup_agents(  # type: ignore[override]
         model = self.get_model_adapter(model_id, register_name="agent_model", seed=agent_seed)
 
         agent = DefaultGaia2Agent(
-            tools=tools,  # type: ignore[arg-type]  # AREToolWrapper is Callable
+            tools=tools,  # type: ignore[arg-type]  # Gaia2GenericTool has __call__
             model=model,
             environment=environment,
             llm_args=llm_args,
diff --git a/tests/test_benchmarks/test_gaia2/test_integration.py b/tests/test_benchmarks/test_gaia2/test_integration.py
@@ -159,7 +159,7 @@ def test_agent_builds_system_prompt_with_real_tools(self, first_real_task):
             )
 
             agent = DefaultGaia2Agent(
-                tools=tools,
+                tools=tools,  # type: ignore[arg-type]  # Gaia2GenericTool has __call__
                 model=model,
                 environment=env,
                 max_iterations=1,

Original file line number	Diff line number	Diff line change
`@@ -159,7 +159,7 @@ def test_agent_builds_system_prompt_with_real_tools(self, first_real_task):`
`159`	`159`	`)`
`160`	`160`
`161`	`161`	`agent = DefaultGaia2Agent(`
`162`		`- tools=tools,`
	`162`	`+ tools=tools, # type: ignore[arg-type] # Gaia2GenericTool has __call__`
`163`	`163`	`model=model,`
`164`	`164`	`environment=env,`
`165`	`165`	`max_iterations=1,`