diff --git a/dataset.py b/dataset.py index b63174e..44a82d8 100644 --- a/dataset.py +++ b/dataset.py @@ -36,7 +36,6 @@ class Comment: class Selection: comment_suggests_change: bool diff_after_address_change: Optional[bool] - is_code_related: bool @dataclass class Metadata: diff --git a/manual_selection.py b/manual_selection.py index 76a8e4f..2703bbc 100644 --- a/manual_selection.py +++ b/manual_selection.py @@ -190,23 +190,21 @@ def main( display_pr_info(entry, i, total, n_good) - is_code_related = any(file.file.endswith('.java') for file in entry.comments) suggests = prompt_comment_suggestion(entry, sel, overwrite) if not suggests: - entry.metadata.selection = Selection(False, None, is_code_related) + entry.metadata.selection = Selection(False, None) continue if validation_mode == ValidationMode.COMMENT: entry.metadata.selection = Selection( True, sel.diff_after_address_change if sel is not None else None, - is_code_related, ) n_good += 1 elif validation_mode == ValidationMode.REFINEMENT: diff_relevant = refine_entry(entry, sel, overwrite, check_diff_relevance) - entry.metadata.selection = Selection(True, diff_relevant, is_code_related) + entry.metadata.selection = Selection(True, diff_relevant) if diff_relevant: n_good += 1 except KeyboardInterrupt: