From dd38554b70d027f7aede205f1ba3ed30d09d124b Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 9 Feb 2025 21:47:50 +0000 Subject: [PATCH] fix: update type hints and imports for defaultdict Co-Authored-By: Joe Moura --- .../utilities/evaluators/crew_evaluator_handler.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/crewai/utilities/evaluators/crew_evaluator_handler.py b/src/crewai/utilities/evaluators/crew_evaluator_handler.py index 3cc6e76d9..d5bf9058d 100644 --- a/src/crewai/utilities/evaluators/crew_evaluator_handler.py +++ b/src/crewai/utilities/evaluators/crew_evaluator_handler.py @@ -1,5 +1,5 @@ -from collections.abc import Callable -from typing import Any, Dict, List, Union, Annotated, DefaultDict +from collections import defaultdict +from typing import Any, Dict, List, Union from pydantic import ( BaseModel, @@ -40,12 +40,12 @@ class CrewEvaluator(BaseModel): llm: Union[str, InstanceOf[LLM], Any] = Field( description="Language model that will run the evaluation." ) - tasks_scores: DefaultDict[int, List[float]] = Field( - default_factory=lambda: DefaultDict(list), + tasks_scores: Dict[int, List[float]] = Field( + default_factory=lambda: defaultdict(list), description="Dictionary to store the scores of the agents for each task." ) - run_execution_times: DefaultDict[int, List[int]] = Field( - default_factory=lambda: DefaultDict(list), + run_execution_times: Dict[int, List[int]] = Field( + default_factory=lambda: defaultdict(list), description="Dictionary to store execution times for each run." ) iteration: int = Field(