tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

post-rebase lint

Browse Source
This commit is contained in:
Wing Lian
2025-05-28 10:35:43 -04:00
parent 3cffe881bb
commit 73a84ad0dd
2 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/core/training_args_base.py
View File

@@ -7,8 +7,6 @@ from typing import Optional
from PIL.Image import Resampling from PIL.Image import Resampling
from axolotl.utils.schemas.enums import RingAttnFunc
@dataclass @dataclass
class AxolotlTrainingMixins: class AxolotlTrainingMixins:

4
src/axolotl/integrations/base.py
View File

@@ -22,7 +22,6 @@ from __future__ import annotations
import collections import collections
import importlib import importlib
import logging
import traceback import traceback
from typing import TYPE_CHECKING, Callable, OrderedDict, Union from typing import TYPE_CHECKING, Callable, OrderedDict, Union
@@ -165,7 +164,6 @@ class BasePlugin:
trainer: The trainer object for training. trainer: The trainer object for training.
""" """
def get_training_args(self, cfg: DictDefault): # pylint: disable=unused-argument): def get_training_args(self, cfg: DictDefault): # pylint: disable=unused-argument):
""" """
Returns custom training arguments to set on TrainingArgs. Returns custom training arguments to set on TrainingArgs.
@@ -178,7 +176,7 @@ class BasePlugin:
""" """
def get_collator_cls_and_kwargs( def get_collator_cls_and_kwargs(
self, cfg: DictDefault, is_eval: bool=False self, cfg: DictDefault, is_eval: bool = False
): # pylint: disable=unused-argument): ): # pylint: disable=unused-argument):
""" """
Returns a custom class for the collator. Returns a custom class for the collator.