From 8154d26614439f287ce44ff26e3157d525068d1e Mon Sep 17 00:00:00 2001 From: Sunny Liu Date: Tue, 8 Apr 2025 16:06:35 -0400 Subject: [PATCH] nit --- src/axolotl/common/datasets.py | 2 +- src/axolotl/prompt_strategies/base.py | 10 +++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/src/axolotl/common/datasets.py b/src/axolotl/common/datasets.py index 627f08c60..3ecbd5b83 100644 --- a/src/axolotl/common/datasets.py +++ b/src/axolotl/common/datasets.py @@ -130,7 +130,7 @@ def load_preference_datasets( if cli_args.debug or cfg.debug: if cfg.rl == "grpo": - LOG.info("skip check_dataset_labels during debug for grpo") + pass else: LOG.info("check_dataset_labels...") diff --git a/src/axolotl/prompt_strategies/base.py b/src/axolotl/prompt_strategies/base.py index e142b5ba2..0e6326b28 100644 --- a/src/axolotl/prompt_strategies/base.py +++ b/src/axolotl/prompt_strategies/base.py @@ -9,7 +9,15 @@ import sys LOG = logging.getLogger("axolotl") -def import_from_path(module_name, file_path): +def import_from_path(module_name: str, file_path: str): + """ + Import a module from a file path. + Args: + module_name (str): Name of the module. + file_path (str): Path to the file. + Feturns: + module: The imported module. + """ spec = importlib.util.spec_from_file_location(module_name, file_path) if spec is None: raise ImportError(f"Could not create module spec for: {file_path}")