From 2afeebe5b000dd99371744408fdc40f30e2143a3 Mon Sep 17 00:00:00 2001 From: zigzagcai Date: Fri, 15 Dec 2023 19:58:22 +0800 Subject: [PATCH] fix pylint --- internlm/train/training_internlm.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/internlm/train/training_internlm.py b/internlm/train/training_internlm.py index db18951..0118748 100644 --- a/internlm/train/training_internlm.py +++ b/internlm/train/training_internlm.py @@ -263,8 +263,8 @@ def get_train_data_loader(num_worker: int = 0, dataset_generate_func: Optional[C data_world_size=gpc.get_world_size(ParallelMode.DATA), ) train_collate_fn = partial(packed_collate_fn, packed_length=data_cfg.packed_length) - - def dl_worker_init(worker_id): + + def dl_worker_init(worker_id): # pylint: disable=unused-argument gc.enable() # Create the training data loader @@ -275,7 +275,7 @@ def get_train_data_loader(num_worker: int = 0, dataset_generate_func: Optional[C pin_memory=True, collate_fn=train_collate_fn, persistent_workers=num_worker > 0, - worker_init_fn = dl_worker_init, + worker_init_fn=dl_worker_init, ) return train_dl, dataset_types