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 output to manual_selection in order not to

Browse Source 
overwrite every time
This commit is contained in:
Karma Riuk
2025-05-14 09:16:08 +02:00
parent 2f2cbae756
commit ea3a2b72e5
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
manual_selection.py
View File

@ -77,6 +77,7 @@ def edit_hunk(hunk: str) -> str:
def main(
dataset_path: str,
output: str,
overwrite: bool = False,
validation_mode: ValidationMode = ValidationMode.REFINEMENT,
check_diff_relevance: bool = False,
@ -230,8 +231,8 @@ def main(
except KeyboardInterrupt:
print("\nInterrupted.")
finally:
print(f"Saving dataset to {dataset_path}...", end=" ", flush=True)
dataset.to_json(dataset_path)
print(f"Saving dataset to {output}...", end=" ", flush=True)
dataset.to_json(output)
print("Done")
@ -239,7 +240,16 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Manual selection of dataset")
parser.add_argument("dataset", type=str, help="Path to the dataset file")
parser.add_argument(
"-o", "--overwrite", action="store_true", help="Re-evaluate existing selections"
"-o",
"--output",
required=True,
type=str,
help="The path to the resulting dataset",
)
parser.add_argument(
"--overwrite",
action="store_true",
help="Re-evaluate existing selections",
)
parser.add_argument(
"-m",