refactor: add override decorator

Author: Lawhy

src/strands_env/eval/aime.py

@@ -18,7 +18,7 @@
 
 import logging
 from collections.abc import Iterable
-from typing import Literal
+from typing import Literal, override
 
 from datasets import load_dataset
 
@@ -49,6 +49,7 @@
 class AIMEEvaluator(Evaluator):
     """Evaluator for AIME math competition problems."""
 
+    @override
     def load_dataset(self, version: Literal["2024", "2025"] = "2024") -> Iterable[Action]:
         """Load AIME dataset from HuggingFace."""
         logger.info(f"Loading AIME {version} dataset from: {AIME_HF_PATHS[version]['path']}")

src/strands_env/eval/evaluator.py

@@ -80,9 +80,8 @@ def __init__(
         self.results: dict[str, list[EvalSample]] = defaultdict(list)
         self.completed_ids: set[str] = set()  # Tracks individual sample IDs for checkpoint
 
-    def load_dataset(self, dataset_path: Path | str) -> Iterable[Action]:
+    def load_dataset(self) -> Iterable[Action]:
         """Load dataset from file. Override to implement custom dataset loading logic."""
-        logger.info(f"Loading dataset from: {dataset_path}")
         raise NotImplementedError("Evaluator subclasses must implement load_dataset()")
 
     def load_results(self) -> None: