From 4009a2ba5f481ffa6bf865231554cd6c0886957e Mon Sep 17 00:00:00 2001 From: lhl Date: Fri, 14 Nov 2025 14:18:55 +0900 Subject: [PATCH] reordered our tests to mirror llm_compressor for prepare_plugins/validate order --- tests/e2e/test_llama4_moe_aux_free.py | 2 +- tests/e2e/test_moe_aux_free.py | 2 +- tests/e2e/test_moe_aux_parity.py | 2 +- tests/e2e/test_qwen3_moe_aux_free.py | 2 +- tests/e2e/test_ring_moe_aux_free.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/e2e/test_llama4_moe_aux_free.py b/tests/e2e/test_llama4_moe_aux_free.py index bceb55816..9385f7733 100644 --- a/tests/e2e/test_llama4_moe_aux_free.py +++ b/tests/e2e/test_llama4_moe_aux_free.py @@ -55,9 +55,9 @@ class TestLlama4MoeAuxFree(unittest.TestCase): } ) + prepare_plugins(cfg) cfg = validate_config(cfg) normalize_config(cfg) - prepare_plugins(cfg) dataset_meta = load_datasets(cfg=cfg) model, _, _ = train(cfg=cfg, dataset_meta=dataset_meta) diff --git a/tests/e2e/test_moe_aux_free.py b/tests/e2e/test_moe_aux_free.py index 266fcc633..237ca490c 100644 --- a/tests/e2e/test_moe_aux_free.py +++ b/tests/e2e/test_moe_aux_free.py @@ -57,9 +57,9 @@ class TestMoeAuxFree(unittest.TestCase): } ) + prepare_plugins(cfg) cfg = validate_config(cfg) normalize_config(cfg) - prepare_plugins(cfg) dataset_meta = load_datasets(cfg=cfg) model, _, _ = train(cfg=cfg, dataset_meta=dataset_meta) diff --git a/tests/e2e/test_moe_aux_parity.py b/tests/e2e/test_moe_aux_parity.py index 11b5a920b..d048bacdd 100644 --- a/tests/e2e/test_moe_aux_parity.py +++ b/tests/e2e/test_moe_aux_parity.py @@ -71,9 +71,9 @@ class TestMoeAuxParity(unittest.TestCase): cfg1.moe_update_rate = 0.01 cfg1.moe_update_momentum = 0.9 cfg1.moe_bias_cap = 2.0 + prepare_plugins(cfg1) cfg1 = validate_config(cfg1) normalize_config(cfg1) - prepare_plugins(cfg1) dataset_meta1 = load_datasets(cfg=cfg1) model1, _, trainer1 = train(cfg=cfg1, dataset_meta=dataset_meta1) loss1 = _last_logged_loss(trainer1) diff --git a/tests/e2e/test_qwen3_moe_aux_free.py b/tests/e2e/test_qwen3_moe_aux_free.py index 204ec4ad3..8f968e9ff 100644 --- a/tests/e2e/test_qwen3_moe_aux_free.py +++ b/tests/e2e/test_qwen3_moe_aux_free.py @@ -55,9 +55,9 @@ class TestQwen3MoeAuxFree(unittest.TestCase): } ) + prepare_plugins(cfg) cfg = validate_config(cfg) normalize_config(cfg) - prepare_plugins(cfg) dataset_meta = load_datasets(cfg=cfg) model, _, _ = train(cfg=cfg, dataset_meta=dataset_meta) diff --git a/tests/e2e/test_ring_moe_aux_free.py b/tests/e2e/test_ring_moe_aux_free.py index 992f703e4..8bfb924a6 100644 --- a/tests/e2e/test_ring_moe_aux_free.py +++ b/tests/e2e/test_ring_moe_aux_free.py @@ -56,9 +56,9 @@ class TestRingMoeAuxFree(unittest.TestCase): } ) + prepare_plugins(cfg) cfg = validate_config(cfg) normalize_config(cfg) - prepare_plugins(cfg) dataset_meta = load_datasets(cfg=cfg) model, _, _ = train(cfg=cfg, dataset_meta=dataset_meta)