karma/crab
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab.git synced 2025-07-04 21:28:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

added enumchoiceaction for easier enum in argparse

Browse Source 
handling
This commit is contained in:
Karma Riuk
2025-05-16 19:41:15 +02:00
parent 14e64984c5
commit 5cf5e5a8ee
2 changed files with 26 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
manual_selection.py
View File

@ -2,7 +2,7 @@ from typing import Optional
from dataset import Dataset, DatasetEntry, Selection
import argparse, os, re, click
from enum import Enum
from utils import prompt_yes_no
from utils import EnumChoicesAction, prompt_yes_no
HUNK_HEADER_REGEX = re.compile(r'^@@ -\d+(?:,\d+)? \+\d+(?:,\d+)? @@')
@ -235,9 +235,10 @@ if __name__ == "__main__":
parser.add_argument(
"-m",
"--mode",
choices=[mode.value for mode in ValidationMode],
default='comment',
help="Validation mode: 'comment' to only check if comments suggest changes, 'refinement' to check both comment suggestions and implementation. Default is 'comment'",
type=ValidationMode,
default=ValidationMode.COMMENT,
action=EnumChoicesAction,
help=f"Validation mode: '{ValidationMode.COMMENT.value}' to only check if comments suggest changes, '{ValidationMode.REFINEMENT.value}' to check both comment suggestions and implementation. Default is '{ValidationMode.COMMENT.value}'",
)
parser.add_argument(
"--check-diff-relevance",
@ -249,6 +250,6 @@ if __name__ == "__main__":
args.dataset,
args.output,
overwrite=args.overwrite,
validation_mode=ValidationMode(args.mode),
validation_mode=args.mode,
check_diff_relevance=args.check_diff_relevance,
)