Commit f01b3e66 authored by karajan1001's avatar karajan1001 Committed by Julien Chaumond
Browse files

fix #2399 an ImportError in official example (#2400)



* fix #2399 an ImportError in official example

* style
Co-authored-by: default avatarJulien Chaumond <chaumond@gmail.com>
parent 78528742
...@@ -9,7 +9,6 @@ import re ...@@ -9,7 +9,6 @@ import re
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from absl import app, flags, logging from absl import app, flags, logging
from fastprogress import master_bar, progress_bar
from seqeval import metrics from seqeval import metrics
from transformers import ( from transformers import (
...@@ -29,6 +28,12 @@ from transformers import ( ...@@ -29,6 +28,12 @@ from transformers import (
from utils_ner import convert_examples_to_features, get_labels, read_examples_from_file 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( ALL_MODELS = sum(
(tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, RobertaConfig, DistilBertConfig)), () (tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, RobertaConfig, DistilBertConfig)), ()
) )
......
Markdown is supported
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