Skip to content
Snippets Groups Projects
Commit 109e0255 authored by Maja Jablonska's avatar Maja Jablonska
Browse files

Minor fixes to CLI prediction

parent 007b8bcc
Branches
Tags
1 merge request!46Merge COMBO 3.0 into master
......@@ -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:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment