From fee55a7d5afd7dbe0d52dd38f195ce33667a02cd Mon Sep 17 00:00:00 2001 From: Jamie Hardt Date: Wed, 3 Sep 2025 14:56:41 -0700 Subject: [PATCH] Twiddles --- ucsinfer/__main__.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/ucsinfer/__main__.py b/ucsinfer/__main__.py index 5f55fce..661b62a 100644 --- a/ucsinfer/__main__.py +++ b/ucsinfer/__main__.py @@ -54,7 +54,7 @@ def ucsinfer(ctx, verbose, no_model_cache, model): if no_model_cache: logger.info("Model cache inhibited by config") - logger.info("Using model {model}") + logger.info(f"Using model {model}") @ucsinfer.command('recommend') @@ -117,12 +117,12 @@ def recommend(ctx, text, paths, interactive, skip_ucs): @ucsinfer.command('gather') -@click.option('--outfile', default='dataset', show_default=True) +@click.option('--out', default='dataset/', show_default=True) @click.option('--ucs-data', flag_value=True, help="Create a dataset based " "on the UCS category explanations and synonymns (PATHS will " "be ignored.)") @click.argument('paths', nargs=-1) -def gather(paths, outfile, ucs_data): +def gather(paths, out, ucs_data): """ Scan files to build a training dataset @@ -148,6 +148,7 @@ def gather(paths, outfile, ucs_data): catid_list = [cat.catid for cat in ucs] if ucs_data: + logger.info('Creating dataset for UCS categories instead of from PATH') paths = [] for path in paths: @@ -182,7 +183,8 @@ def gather(paths, outfile, ucs_data): else: dataset = build_sentence_class_dataset(scan_metadata(), catid_list) - dataset.save_to_disk(outfile) + logger.info(f"Saving dataset to disk at {out}") + dataset.save_to_disk(out) @ucsinfer.command('finetune')