Unverified Commit c40ecfd7 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Move import (#14787)

parent 7c9c41f4
...@@ -47,7 +47,6 @@ from .file_utils import ( ...@@ -47,7 +47,6 @@ from .file_utils import (
is_remote_url, is_remote_url,
) )
from .generation_tf_utils import TFGenerationMixin from .generation_tf_utils import TFGenerationMixin
from .modelcard import TrainingSummary
from .modeling_tf_outputs import TFSeq2SeqLMOutput from .modeling_tf_outputs import TFSeq2SeqLMOutput
from .tokenization_utils_base import BatchEncoding from .tokenization_utils_base import BatchEncoding
from .utils import logging from .utils import logging
...@@ -940,6 +939,9 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin, TFGenerationMixin, Pu ...@@ -940,6 +939,9 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin, TFGenerationMixin, Pu
dataset: Optional[Union[str, List[str]]] = None, dataset: Optional[Union[str, List[str]]] = None,
dataset_args: 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( training_summary = TrainingSummary.from_keras(
self, self,
keras_history=self.history, keras_history=self.history,
......
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