From 5062eca069780d25b0ec1afa8ba6f04d4a5f0864 Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Mon, 29 May 2023 10:05:11 +0900
Subject: [PATCH] Lint callbacks.py

---
 src/axolotl/utils/callbacks.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/axolotl/utils/callbacks.py b/src/axolotl/utils/callbacks.py
index 229cd9b98..70e83d6e4 100644
--- a/src/axolotl/utils/callbacks.py
+++ b/src/axolotl/utils/callbacks.py
@@ -1,7 +1,8 @@
+"""Callbacks for Trainer class"""
+
 import os
 
 from transformers import (
-    Seq2SeqTrainer,
     TrainerCallback,
     TrainingArguments,
     TrainerState,
@@ -10,7 +11,9 @@ from transformers import (
 from transformers.trainer_utils import PREFIX_CHECKPOINT_DIR
 
 
-class SavePeftModelCallback(TrainerCallback):
+class SavePeftModelCallback(TrainerCallback):  # pylint: disable=too-few-public-methods
+    """Callback to save the PEFT adapter"""
+
     def on_save(
         self,
         args: TrainingArguments,