Skip to content

Commit

Permalink
Update evalllm.py
Browse files Browse the repository at this point in the history
  • Loading branch information
sourabhagr authored Feb 27, 2024
1 parent 1e18cb6 commit 54fae2c
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions uptrain/framework/evalllm.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"""

import typing as t

from datetime import datetime
from loguru import logger
import httpx
import polars as pl
Expand Down Expand Up @@ -76,9 +76,9 @@ def __init__(self, settings: Settings = None, openai_api_key: str = None) -> Non

def evaluate(
self,
project_name: str,
data: t.Union[list[dict], pl.DataFrame, pd.DataFrame],
checks: list[t.Union[str, Evals, ParametricEval]],
project_name: str = "Project - " + str(datetime.utcnow()),
scenario_description: t.Union[str, list[str], None] = None,
schema: t.Union[DataSchema, dict[str, str], None] = None,
metadata: t.Optional[dict[str, str]] = None,
Expand Down Expand Up @@ -388,4 +388,4 @@ def evaluate_prompts(
schema=schema,
metadata=metadata,
)
return results
return results

0 comments on commit 54fae2c

Please sign in to comment.