From cf8174db75a13e77a17267f71806bf43bc5594c3 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 19 Dec 2024 02:04:25 -0500 Subject: [PATCH] chore: lint --- src/axolotl/utils/collators/kd.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/axolotl/utils/collators/kd.py b/src/axolotl/utils/collators/kd.py index 256dbe7a8..155840912 100644 --- a/src/axolotl/utils/collators/kd.py +++ b/src/axolotl/utils/collators/kd.py @@ -23,6 +23,7 @@ class DataCollatorForKD(DataCollatorForSeq2Seq): target_logprobs. It also creates a teacher_mask to indicate which entries are valid. """ + # pylint: disable=duplicate-code tokenizer: PreTrainedTokenizerBase model: Optional[Any] = None padding: Union[bool, str, PaddingStrategy] = True