Move import (huggingface#14787)

sgugger · sgugger · commit d6698ecfc3ec · 2021-12-15T13:53:18.000-05:00
diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py
@@ -47,7 +47,6 @@
     is_remote_url,
 )
 from .generation_tf_utils import TFGenerationMixin
-from .modelcard import TrainingSummary
 from .modeling_tf_outputs import TFSeq2SeqLMOutput
 from .tokenization_utils_base import BatchEncoding
 from .utils import logging
@@ -940,6 +939,9 @@ def create_model_card(
         dataset: Optional[Union[str, List[str]]] = None,
         dataset_args: Optional[Union[str, List[str]]] = None,
     ):
+        # Avoids a circular import by doing this when necessary.
+        from .modelcard import TrainingSummary
+
         training_summary = TrainingSummary.from_keras(
             self,
             keras_history=self.history,
