Fix minor pytype bug

PiperOrigin-RevId: 700133692 Change-Id: Icd6e7b41f35bea39fe07173aef19676e3efa86a2
google-deepmind · Nov 26, 2024 · a2d48fc · a2d48fc
1 parent 7cbf802
commit a2d48fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/modular/launch_concordia_challenge_evaluation.py b/examples/modular/launch_concordia_challenge_evaluation.py
@@ -67,6 +67,7 @@
 """
 
 import argparse
+from collections.abc import Sequence
 import datetime
 import functools
 import importlib
@@ -230,7 +231,7 @@ def _evaluate_one_repetition(
 def _evaluate_all_repetitions_on_one_scenario(
     scenario_name: str,
     scenario_config: scenarios_lib.ScenarioConfig,
-) -> logging_lib.ScenarioResult:
+) -> Sequence[logging_lib.ScenarioResult]:
   """Evaluates the agent on one scenario, averaging over repetitions.
 
   Args: