From 6491152a623f7a5f199843d7a3f8ce37db852044 Mon Sep 17 00:00:00 2001 From: jjmachan Date: Mon, 13 Jan 2025 19:54:10 +0530 Subject: [PATCH] style: fixed some ci runs --- src/ragas/dataset_schema.py | 8 ++++---- src/ragas/metrics/_domain_specific_rubrics.py | 4 +++- src/ragas/metrics/base.py | 6 +----- 3 files changed, 8 insertions(+), 10 deletions(-) diff --git a/src/ragas/dataset_schema.py b/src/ragas/dataset_schema.py index f8b34e765..ccc246738 100644 --- a/src/ragas/dataset_schema.py +++ b/src/ragas/dataset_schema.py @@ -13,21 +13,21 @@ from datasets import Dataset as HFDataset from pydantic import BaseModel, field_validator +from ragas._version import __version__ from ragas.callbacks import ChainRunEncoder, parse_run_traces from ragas.cost import CostCallbackHandler from ragas.exceptions import UploadException from ragas.messages import AIMessage, HumanMessage, ToolCall, ToolMessage from ragas.sdk import ( - upload_packet, RAGAS_API_SOURCE, - get_app_token, - check_api_response, build_evaluation_app_url, + check_api_response, get_api_url, + get_app_token, get_app_url, + upload_packet, ) from ragas.utils import safe_nanmean -from ragas._version import __version__ if t.TYPE_CHECKING: from pathlib import Path diff --git a/src/ragas/metrics/_domain_specific_rubrics.py b/src/ragas/metrics/_domain_specific_rubrics.py index fbbe24bbb..47d70a0a7 100644 --- a/src/ragas/metrics/_domain_specific_rubrics.py +++ b/src/ragas/metrics/_domain_specific_rubrics.py @@ -113,7 +113,9 @@ def __init__( } # Add rubrics to the scoring prompts - rubrics_text = "\n".join(f"{key}: {value}" for key, value in self.rubrics.items()) + rubrics_text = "\n".join( + f"{key}: {value}" for key, value in self.rubrics.items() + ) self.single_turn_scoring_prompt.instruction = f"{self.single_turn_scoring_prompt.instruction}\n\nScoring Rubrics:\n{rubrics_text}\n" self.multi_turn_scoring_prompt.instruction = f"{self.multi_turn_scoring_prompt.instruction}\n\nScoring Rubrics:\n{rubrics_text}\n" diff --git a/src/ragas/metrics/base.py b/src/ragas/metrics/base.py index 156769c14..04e519ecf 100644 --- a/src/ragas/metrics/base.py +++ b/src/ragas/metrics/base.py @@ -13,11 +13,7 @@ from ragas._analytics import EvaluationEvent, _analytics_batcher from ragas.callbacks import ChainType, new_group -from ragas.dataset_schema import ( - MetricAnnotation, - MultiTurnSample, - SingleTurnSample, -) +from ragas.dataset_schema import MetricAnnotation, MultiTurnSample, SingleTurnSample from ragas.executor import is_event_loop_running from ragas.losses import BinaryMetricLoss, MSELoss from ragas.prompt import FewShotPydanticPrompt, PromptMixin