From 109e02552b6452fb93062162b11a1f5a176b5250 Mon Sep 17 00:00:00 2001 From: Maja Jablonska <majajjablonska@gmail.com> Date: Sat, 18 Nov 2023 18:37:10 +1100 Subject: [PATCH] Minor fixes to CLI prediction --- combo/main.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/combo/main.py b/combo/main.py index 9736646..c815130 100755 --- a/combo/main.py +++ b/combo/main.py @@ -309,9 +309,9 @@ def run(_): if not dataset_reader: logger.info("No dataset reader in the configuration or archive file - using a default UD dataset reader", prefix=prefix) - dataset_reader = default_ud_dataset_reader() + dataset_reader = default_ud_dataset_reader(FLAGS.pretrained_transformer_name) - predictor = COMBO(model, dataset_reader) + predictor = COMBO(model, dataset_reader, LamboTokenizer(language=FLAGS.tokenizer_language)) if FLAGS.input_file == '-': print("Interactive mode.") @@ -322,7 +322,11 @@ def run(_): print("{:15} {:15} {:10} {:10} {:10}".format(token.text, token.lemma, token.upostag, token.head, token.deprel)) elif FLAGS.output_file: - checks.file_exists(FLAGS.input_file) + try: + checks.file_exists(FLAGS.input_file) + except ConfigurationError as e: + handle_error(e, prefix) + logger.info("Predicting examples from file", prefix=prefix) if FLAGS.conllu_format: -- GitLab