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
No related branches found
No related tags found
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:
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% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment