From b708a1cc450617e2ed665543b3a3020554213b63 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sat, 26 Apr 2025 13:11:25 -0400 Subject: [PATCH] validate config to set defaults --- tests/e2e/integrations/test_llm_compressor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/e2e/integrations/test_llm_compressor.py b/tests/e2e/integrations/test_llm_compressor.py index 707beacbf..98e8caf49 100644 --- a/tests/e2e/integrations/test_llm_compressor.py +++ b/tests/e2e/integrations/test_llm_compressor.py @@ -9,7 +9,7 @@ import pytest from axolotl.cli.args import TrainerCliArgs from axolotl.common.datasets import load_datasets from axolotl.train import train -from axolotl.utils.config import normalize_config, prepare_plugins +from axolotl.utils.config import normalize_config, prepare_plugins, validate_config from axolotl.utils.dict import DictDefault from tests.e2e.utils import check_model_output_exists, require_torch_2_4_1 @@ -79,6 +79,7 @@ class TestLLMCompressorIntegration: ) prepare_plugins(cfg) + cfg = validate_config(cfg) normalize_config(cfg) cli_args = TrainerCliArgs() dataset_meta = load_datasets(cfg=cfg, cli_args=cli_args)