show steps num for tape in the tape selector

Author: ollmer

src/agentlab/analyze/tapes.py

@@ -23,7 +23,7 @@ class WrapperStep(Step):
     content: dict
 
 
-def pretty_yaml(data: dict) -> str:
+def pretty_yaml(data: dict | None) -> str:
     return yaml.dump(data, sort_keys=False, indent=2) if data else ""
 
 
@@ -117,7 +117,7 @@ def get_tape_name(self, i: int, tape: Tape) -> str:
         number = tape.metadata.task.get("number", "")
         n = f"{tape.metadata.task.get('Level', '')}.{number} " if number else ""
         name = tape.steps[0].content["content"][:32] + "..."
-        return f"{n}{mark}{name}"
+        return f"{n}({len(tape.steps)}){mark}{name}"
 
     def get_exp_label(self, filename: str, tapes: list[Tape]) -> str:
         acc, n_solved = self.calculate_accuracy(tapes)

src/agentlab/benchmarks/gaia.py

@@ -145,7 +145,7 @@ def model_post_init(self, __context: Any) -> None:
                 name="2023_all",
                 trust_remote_code=True,
             )  # type: ignore
-        for task in self.dataset[self.split]:
+        for task in self.dataset[self.split]:  # type: ignore
             if self.level != "all" and task["Level"] != self.level:
                 continue
             number += 1
@@ -314,7 +314,7 @@ def question_scorer(
     model_answer: str,
     ground_truth: str,
 ) -> bool:
-    def is_float(element: any) -> bool:
+    def is_float(element: Any) -> bool:
         try:
             float(element)
             return True