From 24b61c1b67c80eaec813351ce5a6cb254578432d Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Thu, 22 May 2025 17:44:30 +0700 Subject: [PATCH] chore: lint --- src/axolotl/core/trainer_builder/rl.py | 2 +- src/axolotl/core/trainer_builder/sft.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/core/trainer_builder/rl.py b/src/axolotl/core/trainer_builder/rl.py index 578f68517..f62e14880 100644 --- a/src/axolotl/core/trainer_builder/rl.py +++ b/src/axolotl/core/trainer_builder/rl.py @@ -197,7 +197,7 @@ class HFRLTrainerBuilder(TrainerBuilderBase): trainer_cls, trainer_cls_args = self._get_trainer_cls(trainer_kwargs) sig = inspect.signature(trainer_cls) - if "tokenizer" in sig.parameters.keys(): + if "tokenizer" in sig.parameters: trainer_kwargs["tokenizer"] = self.tokenizer else: trainer_kwargs["processing_class"] = self.tokenizer diff --git a/src/axolotl/core/trainer_builder/sft.py b/src/axolotl/core/trainer_builder/sft.py index 27750df15..3fea0e4d8 100644 --- a/src/axolotl/core/trainer_builder/sft.py +++ b/src/axolotl/core/trainer_builder/sft.py @@ -400,7 +400,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): **data_collator_kwargs, ) sig = inspect.signature(trainer_cls) - if "processing_class" in sig.parameters.keys(): + if "processing_class" in sig.parameters: trainer_kwargs["processing_class"] = self.tokenizer else: trainer_kwargs["tokenizer"] = self.tokenizer