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

fixed log statements

Browse Source
This commit is contained in:
Karma Riuk
2025-05-07 10:32:02 +02:00
parent 40fa958cf8
commit 97646cb8c3
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dataset.py
View File

@ -111,6 +111,7 @@ class Dataset:
] ]
to_dump = Dataset(entries=entries_to_dump) to_dump = Dataset(entries=entries_to_dump)
print(f"{len(entries_to_dump)} entries...", end=" ", flush=True)
def transform_entry(entry: DatasetEntry | Dataset | Any) -> dict | list: def transform_entry(entry: DatasetEntry | Dataset | Any) -> dict | list:
if not isinstance(entry, (DatasetEntry, Dataset)): if not isinstance(entry, (DatasetEntry, Dataset)):
@ -196,7 +197,7 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
dataset = Dataset.from_json(args.filename) dataset = Dataset.from_json(args.filename)
print(f"Loaded {len(dataset)} entries from {args.filename}") print(f"Loaded {len(dataset.entries)} entries from {args.filename}")
if os.path.exists(args.output): if os.path.exists(args.output):
overwrite = prompt_yes_no( overwrite = prompt_yes_no(
f"Output file {args.output} already exists. Do you want to overwrite it?" f"Output file {args.output} already exists. Do you want to overwrite it?"
@ -204,6 +205,6 @@ if __name__ == "__main__":
if not overwrite: if not overwrite:
print("Exiting without saving.") print("Exiting without saving.")
exit(0) exit(0)
print(f"Saving dataset to {args.output}...", end=" ", flush=True) print(f"Saving dataset to {args.output},", end=" ", flush=True)
dataset.to_json(args.output, OutputType(args.output_type)) dataset.to_json(args.output, OutputType(args.output_type))
print("Done") print("Done")