[CodeStyle] Format trainer code style. (#4057)

Author: ZHUI

paddlenlp/trainer/compression_args.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from dataclasses import dataclass, field
 import types
+from dataclasses import dataclass, field
 from typing import List, Optional
 
 import paddle

paddlenlp/trainer/integrations.py

@@ -19,9 +19,9 @@
 import importlib
 import json
 
-from .trainer_callback import TrainerCallback
-from ..utils.log import logger
 from ..transformers import PretrainedModel
+from ..utils.log import logger
+from .trainer_callback import TrainerCallback
 
 
 def is_visualdl_available():

paddlenlp/trainer/trainer_callback.py

@@ -26,9 +26,10 @@
 import numpy as np
 from tqdm.auto import tqdm
 
+from paddlenlp.utils.log import logger
+
 from .trainer_utils import IntervalStrategy, has_length
 from .training_args import TrainingArguments
-from paddlenlp.utils.log import logger
 
 __all__ = [
     "TrainerState",

paddlenlp/trainer/trainer_seq2seq.py

@@ -20,7 +20,6 @@
 
 from .trainer import Trainer
 from .trainer_utils import PredictionOutput
-from ..utils.log import logger
 
 __all__ = [
     "Seq2SeqTrainer",

paddlenlp/trainer/trainer_utils.py

@@ -16,25 +16,25 @@
 # This file is modified from
 #  https://github.com/huggingface/transformers/blob/main/src/transformers/trainer_utils.py
 """
-Utilities for the Trainer class. 
+Utilities for the Trainer class.
 """
 import datetime
 import json
+import math
 import os
 import random
 import re
 import time
-import math
 from enum import Enum
-from typing import Dict, NamedTuple, Optional, Tuple, Union, List
+from typing import Dict, List, NamedTuple, Optional, Tuple, Union
 
 import numpy as np
 import paddle
 from paddle.io import IterableDataset
 from paddle.optimizer.lr import LambdaDecay
 
-from ..utils.log import logger
 from ..transformers.tokenizer_utils_base import BatchEncoding
+from ..utils.log import logger
 
 __all__ = [
     "TrainOutput",
@@ -459,18 +459,6 @@ def has_length(dataset):
         return False
 
 
-def get_last_checkpoint(folder):
-    content = os.listdir(folder)
-    checkpoints = [
-        path
-        for path in content
-        if _re_checkpoint.search(path) is not None and os.path.isdir(os.path.join(folder, path))
-    ]
-    if len(checkpoints) == 0:
-        return
-    return os.path.join(folder, max(checkpoints, key=lambda x: int(_re_checkpoint.search(x).groups()[0])))
-
-
 class IterableDatasetShard(IterableDataset):
     """
     Wraps a Paddle `IterableDataset` to generate samples for one of the processes only. Instances of this class will

paddlenlp/trainer/training_args.py

@@ -20,20 +20,20 @@
 import json
 import math
 import os
-from dataclasses import asdict, dataclass, field
+import types
 import warnings
+from dataclasses import asdict, dataclass, field
 from enum import Enum
-import types
 from typing import Any, Dict, List, Optional
 
 import paddle
 from paddle.distributed import fleet
 
 from ..utils.log import logger
 from .trainer_utils import (
-    SchedulerType,
     IntervalStrategy,
     OptimizerNames,
+    SchedulerType,
     ShardingOption,
 )
 

paddlenlp/trainer/training_args_seq2seq.py

@@ -12,15 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import logging
 from dataclasses import dataclass, field
 from typing import Optional
 
 from .training_args import TrainingArguments
 from .utils import add_start_docstrings
 
-from ..utils.log import logger
-
 __all__ = [
     "Seq2SeqTrainingArguments",
 ]