Skip to content

Commit a7810c5

Browse files
committed
Import protection.
1 parent ab430c8 commit a7810c5

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

src/transformers/pipelines/question_answering.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,17 @@
55

66
import numpy as np
77

8-
import tokenizers
9-
108
from ..data import SquadExample, SquadFeatures, squad_convert_examples_to_features
119
from ..modelcard import ModelCard
1210
from ..tokenization_utils import PreTrainedTokenizer
13-
from ..utils import PaddingStrategy, add_end_docstrings, is_tf_available, is_torch_available, logging
11+
from ..utils import (
12+
PaddingStrategy,
13+
add_end_docstrings,
14+
is_tf_available,
15+
is_tokenizers_available,
16+
is_torch_available,
17+
logging,
18+
)
1419
from .base import PIPELINE_INIT_ARGS, ArgumentHandler, ChunkPipeline
1520

1621

@@ -20,6 +25,9 @@
2025
from ..modeling_tf_utils import TFPreTrainedModel
2126
from ..modeling_utils import PreTrainedModel
2227

28+
if is_tokenizers_available():
29+
import tokenizers
30+
2331
if is_tf_available():
2432
import tensorflow as tf
2533

0 commit comments

Comments
 (0)