From 73a84ad0dd8f6736b8c1c16912e26bf15e2865c0 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 28 May 2025 10:35:43 -0400 Subject: [PATCH] post-rebase lint --- src/axolotl/core/training_args_base.py | 2 -- src/axolotl/integrations/base.py | 4 +--- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/src/axolotl/core/training_args_base.py b/src/axolotl/core/training_args_base.py index fbd387e5f..a158bc4a8 100644 --- a/src/axolotl/core/training_args_base.py +++ b/src/axolotl/core/training_args_base.py @@ -7,8 +7,6 @@ from typing import Optional from PIL.Image import Resampling -from axolotl.utils.schemas.enums import RingAttnFunc - @dataclass class AxolotlTrainingMixins: diff --git a/src/axolotl/integrations/base.py b/src/axolotl/integrations/base.py index 001436152..91b380138 100644 --- a/src/axolotl/integrations/base.py +++ b/src/axolotl/integrations/base.py @@ -22,7 +22,6 @@ from __future__ import annotations import collections import importlib -import logging import traceback from typing import TYPE_CHECKING, Callable, OrderedDict, Union @@ -165,7 +164,6 @@ class BasePlugin: trainer: The trainer object for training. """ - def get_training_args(self, cfg: DictDefault): # pylint: disable=unused-argument): """ Returns custom training arguments to set on TrainingArgs. @@ -178,7 +176,7 @@ class BasePlugin: """ def get_collator_cls_and_kwargs( - self, cfg: DictDefault, is_eval: bool=False + self, cfg: DictDefault, is_eval: bool = False ): # pylint: disable=unused-argument): """ Returns a custom class for the collator.