File tree Expand file tree Collapse file tree 2 files changed +19
-2
lines changed
Expand file tree Collapse file tree 2 files changed +19
-2
lines changed Original file line number Diff line number Diff line change @@ -123,6 +123,17 @@ def __init__(self, tasks_dir) -> None:
123123 self .tasks_dir = tasks_dir
124124 self .message = f"Invalid Tasks Dir: { tasks_dir } "
125125
126+ class InvalidEvaluationResult (EvalError ):
127+ """
128+ Error raised for invalid eval results
129+ Attributes
130+ message error message to be printed on raise
131+ """
132+
133+ def __init__ (self , message ) -> None :
134+ super ().__init__ ()
135+ self .message = message
136+
126137
127138class ModelServingAPIError (EvalError ):
128139 """
Original file line number Diff line number Diff line change 88import numpy as np
99import pandas as pd
1010
11+ # First Party
12+ from instructlab .eval import exceptions
13+
1114# Local
1215from .logger_config import setup_logger
1316from .mt_bench_common import (
@@ -97,8 +100,11 @@ def make_judgment(
97100 turn_scores = []
98101 # First turn
99102 df_1 = judgment_df [judgment_df ["turn" ] == 1 ].groupby (["model" , "turn" ]).mean ()
100- overall_score = df_1 ["score" ].iloc [0 ]
101- turn_scores .append (overall_score )
103+ if len (df_1 .index ) > 0 :
104+ overall_score = df_1 ["score" ].iloc [0 ]
105+ turn_scores .append (overall_score )
106+ else :
107+ raise exceptions .InvalidEvaluationResult ("Evaluation provided no result. See logs for more details." )
102108
103109 if bench_name == "mt_bench" :
104110 # Second turn
You can’t perform that action at this time.
0 commit comments