Commit d0c9fe27 authored by Aymeric Augustin's avatar Aymeric Augustin
Browse files

Fix circular import in transformers.pipelines.

Submodules shouldn't import from their parent in general.
parent 5ca05475
...@@ -29,24 +29,19 @@ from typing import Dict, List, Optional, Tuple, Union ...@@ -29,24 +29,19 @@ from typing import Dict, List, Optional, Tuple, Union
import numpy as np import numpy as np
import six import six
from transformers import ( from .configuration_auto import ALL_PRETRAINED_CONFIG_ARCHIVE_MAP, AutoConfig
ALL_PRETRAINED_CONFIG_ARCHIVE_MAP, from .configuration_utils import PretrainedConfig
AutoConfig, from .data import SquadExample, squad_convert_examples_to_features
AutoTokenizer, from .file_utils import is_tf_available, is_torch_available
BasicTokenizer, from .modelcard import ModelCard
ModelCard, from .tokenization_auto import AutoTokenizer
PretrainedConfig, from .tokenization_bert import BasicTokenizer
PreTrainedTokenizer, from .tokenization_utils import PreTrainedTokenizer
SquadExample,
is_tf_available,
is_torch_available,
squad_convert_examples_to_features,
)
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers import ( from .modeling_tf_auto import (
TFAutoModel, TFAutoModel,
TFAutoModelForSequenceClassification, TFAutoModelForSequenceClassification,
TFAutoModelForQuestionAnswering, TFAutoModelForQuestionAnswering,
...@@ -55,7 +50,7 @@ if is_tf_available(): ...@@ -55,7 +50,7 @@ if is_tf_available():
if is_torch_available(): if is_torch_available():
import torch import torch
from transformers import ( from .modeling_auto import (
AutoModel, AutoModel,
AutoModelForSequenceClassification, AutoModelForSequenceClassification,
AutoModelForQuestionAnswering, AutoModelForQuestionAnswering,
......
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