From a82af17e07c5423f2f8b8e6623a1680a2c066316 Mon Sep 17 00:00:00 2001 From: Brad Hilton Date: Fri, 25 Sep 2026 03:57:42 +0000 Subject: [PATCH] Run the MoE planner memory tests in the Megatron CI lane The root unit-test environment lacks Megatron, so importorskip skipped 28 of the module's tests there. Co-Authored-By: Claude Opus 5.5 (1M context) --- .github/workflows/prek.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/prek.yml b/.github/workflows/prek.yml index 17e6913da..453b7f8b2 100644 --- a/.github/workflows/prek.yml +++ b/.github/workflows/prek.yml @@ -231,6 +231,7 @@ jobs: tests/unit/test_trainer_rank_checkpoint_memory.py \ tests/unit/test_trainer_rank_checkpoint_gradient_memory.py \ tests/unit/test_trainer_rank_slot_memory.py \ + tests/unit/test_trainer_rank_moe_memory.py \ tests/unit/test_trainer_rank_head_memory.py \ tests/unit/test_trainer_rank_mixed_head_memory.py \ tests/unit/test_trainer_rank_ignored_mixed_head.py \ @@ -272,6 +273,7 @@ jobs: --ignore=tests/unit/test_trainer_rank_checkpoint_memory.py \ --ignore=tests/unit/test_trainer_rank_checkpoint_gradient_memory.py \ --ignore=tests/unit/test_trainer_rank_slot_memory.py \ + --ignore=tests/unit/test_trainer_rank_moe_memory.py \ --ignore=tests/unit/test_trainer_rank_head_memory.py \ --ignore=tests/unit/test_trainer_rank_mixed_head_memory.py \ --ignore=tests/unit/test_trainer_rank_ignored_mixed_head.py \