fix pylint

2023-12-15 19:58:22 +08:00 · 2023-12-15 19:58:22 +08:00 · 2afeebe5b0
parent 2dc8ddd582
commit 2afeebe5b0
1 changed files with 3 additions and 3 deletions
--- a/internlm/train/training_internlm.py
+++ b/internlm/train/training_internlm.py
@ -264,7 +264,7 @@ def get_train_data_loader(num_worker: int = 0, dataset_generate_func: Optional[C
        )
        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