From f1204c4c356789648697d932539368fd24a713f7 Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Thu, 12 Jun 2025 13:25:41 +0200 Subject: [PATCH] fixed extraction of correct predictions --- extract_correct_predictions.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/extract_correct_predictions.py b/extract_correct_predictions.py index 8975eaa..d8ad697 100644 --- a/extract_correct_predictions.py +++ b/extract_correct_predictions.py @@ -72,7 +72,11 @@ def extract_refinement_predictions(dataset_path: str, archives_path: str, output # Iterate over entries that address the change for entry in tqdm(dataset.entries, unit="entries"): sel = entry.metadata.selection - if not sel or not (sel.diff_after_address_change and entry.metadata.is_code_related): + if not sel or not ( + sel.diff_after_address_change + and entry.metadata.is_code_related + and entry.metadata.is_covered + ): continue entry_id = entry.metadata.id