Skip to content

Commit d1075af

Browse files
committed
fix(evaluate): Display table
1 parent 0c14d38 commit d1075af

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dspy/evaluate/evaluate.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -218,12 +218,12 @@ def wrapped_program(example_idx, example):
218218
result_df = result_df.rename(columns={"correct": metric_name})
219219

220220
if display_table:
221-
if isinstance(display_table, int):
222-
df_to_display = result_df.head(display_table).copy()
223-
truncated_rows = len(result_df) - display_table
224-
else:
221+
if isinstance(display_table, bool):
225222
df_to_display = result_df.copy()
226223
truncated_rows = 0
224+
else:
225+
df_to_display = result_df.head(display_table).copy()
226+
truncated_rows = len(result_df) - display_table
227227

228228
styled_df = configure_dataframe_display(df_to_display, metric_name)
229229

0 commit comments

Comments
 (0)