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 \