From de425450d4cedbae0f5db2ab42ecab0be03b71d6 Mon Sep 17 00:00:00 2001 From: Lucas Gomide Date: Fri, 11 Jul 2025 00:09:49 -0300 Subject: [PATCH] style: fix linter issues --- src/crewai/evaluation/__init__.py | 2 +- src/crewai/evaluation/experiment/result.py | 2 +- src/crewai/evaluation/experiment/result_display.py | 2 +- src/crewai/evaluation/experiment/runner.py | 3 +-- tests/evaluation/test_experiment_runner.py | 8 +------- 5 files changed, 5 insertions(+), 12 deletions(-) diff --git a/src/crewai/evaluation/__init__.py b/src/crewai/evaluation/__init__.py index 3f1d8b4df..bdcdce76d 100644 --- a/src/crewai/evaluation/__init__.py +++ b/src/crewai/evaluation/__init__.py @@ -58,5 +58,5 @@ __all__ = [ "create_default_evaluator", "ExperimentRunner", "ExperimentResults", - "TestCaseResult" + "ExperimentResult" ] \ No newline at end of file diff --git a/src/crewai/evaluation/experiment/result.py b/src/crewai/evaluation/experiment/result.py index 5e076cf74..eb6c451ac 100644 --- a/src/crewai/evaluation/experiment/result.py +++ b/src/crewai/evaluation/experiment/result.py @@ -1,7 +1,7 @@ import json import os from datetime import datetime, timezone -from typing import Any, Dict, Optional +from typing import Any from pydantic import BaseModel class ExperimentResult(BaseModel): diff --git a/src/crewai/evaluation/experiment/result_display.py b/src/crewai/evaluation/experiment/result_display.py index bc6c21609..81d60f5f7 100644 --- a/src/crewai/evaluation/experiment/result_display.py +++ b/src/crewai/evaluation/experiment/result_display.py @@ -2,7 +2,7 @@ from typing import Dict, Any from rich.console import Console from rich.table import Table from rich.panel import Panel -from crewai.evaluation.experiment.result import ExperimentResult, ExperimentResults +from crewai.evaluation.experiment.result import ExperimentResults class ExperimentResultsDisplay: def __init__(self): diff --git a/src/crewai/evaluation/experiment/runner.py b/src/crewai/evaluation/experiment/runner.py index 99571adc0..6422a772f 100644 --- a/src/crewai/evaluation/experiment/runner.py +++ b/src/crewai/evaluation/experiment/runner.py @@ -1,7 +1,6 @@ from collections import defaultdict from hashlib import md5 -from typing import List, Dict, Union, Optional, Any -from rich.console import Console +from typing import List, Dict, Union, Any from crewai import Crew from crewai.evaluation import AgentEvaluator, create_default_evaluator diff --git a/tests/evaluation/test_experiment_runner.py b/tests/evaluation/test_experiment_runner.py index a02ce504b..52eefeef5 100644 --- a/tests/evaluation/test_experiment_runner.py +++ b/tests/evaluation/test_experiment_runner.py @@ -1,15 +1,9 @@ import pytest from unittest.mock import MagicMock, patch -import json -from crewai import llm -from crewai.agent import Agent -from crewai.evaluation import metrics -from crewai.task import Task from crewai.crew import Crew from crewai.evaluation.experiment.runner import ExperimentRunner -from crewai.evaluation.experiment.result import ExperimentResult, ExperimentResults -from crewai.evaluation.agent_evaluator import AgentEvaluator +from crewai.evaluation.experiment.result import ExperimentResults from crewai.evaluation.evaluation_display import AgentAggregatedEvaluationResult from crewai.evaluation.base_evaluator import MetricCategory, EvaluationScore