small rename

This commit is contained in:
Ashwin Bharambe 2024-07-11 00:07:19 -07:00
parent 631328f556
commit 86c2993296
3 changed files with 39 additions and 46 deletions

View file

@ -302,9 +302,7 @@ class KPromptGenerations:
@json_schema_type
@dataclass
class MessageScore:
"""A single message and its score."""
class ScoredMessage:
message: Message
score: float
@ -313,7 +311,7 @@ class MessageScore:
@dataclass
class KScoredPromptGenerations:
prompt: Message
k_scored_generations: List[MessageScore]
k_scored_generations: List[ScoredMessage]
@json_schema_type