diff --git a/fieldExtraction/src/testbed/test.py b/fieldExtraction/src/testbed/test.py index 1c8415d..c5e2c4e 100644 --- a/fieldExtraction/src/testbed/test.py +++ b/fieldExtraction/src/testbed/test.py @@ -133,7 +133,7 @@ for field_type, fields in one_to_n_fields.items(): common_files = set(testbed['FILE_NAME']).intersection(set(results['FILE_NAME'])) reimb_primary_metric_rows = [] for file in common_files: - result = testbed_utils.compare_term_extraction_results(file, testbed, results) + result = testbed_utils.compare_term_extraction_results(file, predictions_df=results, gt_df=testbed) reimb_primary_metric_rows.append(result.copy()) reimb_primary_df = pd.DataFrame(reimb_primary_metric_rows)