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
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)), ()
)
......
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