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