Skip to content

Commit b5f441c

Browse files
authored
removed duplicates in data (#617)
1 parent 1051319 commit b5f441c

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

uptrain/framework/evalllm.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -405,9 +405,12 @@ def evaluate_experiments(
405405
all_cols = set(results.columns)
406406
value_cols = list(all_cols - set([schema.question] + exp_columns))
407407
index_cols = metadata.get("uptrain_index_columns", [schema.question])
408+
if sum(results.is_duplicated()) > 1:
409+
logger.info("Duplicates found in data: Removing duplicates")
410+
results = results.unique()
408411
exp_results = results.pivot(
409-
values=value_cols, index=index_cols, columns=exp_columns
410-
)
412+
values=value_cols, index=index_cols, columns=exp_columns
413+
)
411414
exp_results = exp_results.to_dicts()
412415
return exp_results
413416

0 commit comments

Comments
 (0)