From f01b3e6680b49dbc373543a571874f40a8555c9c Mon Sep 17 00:00:00 2001 From: karajan1001 Date: Mon, 6 Jan 2020 01:50:20 +0800 Subject: [PATCH] fix #2399 an ImportError in official example (#2400) * fix #2399 an ImportError in official example * style Co-authored-by: Julien Chaumond --- examples/run_tf_ner.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/examples/run_tf_ner.py b/examples/run_tf_ner.py index 6aa0f4bc320..ef970d83901 100644 --- a/examples/run_tf_ner.py +++ b/examples/run_tf_ner.py @@ -9,7 +9,6 @@ import re import numpy as np import tensorflow as tf from absl import app, flags, logging -from fastprogress import master_bar, progress_bar from seqeval import metrics from transformers import ( @@ -29,6 +28,12 @@ from transformers import ( from utils_ner import convert_examples_to_features, get_labels, read_examples_from_file +try: + from fastprogress import master_bar, progress_bar +except ImportError: + from fastprogress.fastprogress import master_bar, progress_bar + + ALL_MODELS = sum( (tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, RobertaConfig, DistilBertConfig)), () )