[UT for RL] Add UT to cover release/resume memory case for moe model (#8803)

2025-09-10 10:25:12 +08:00
parent 676a7b51bd
commit dccf52f9c8
4 changed files with 79 additions and 5 deletions
--- a/test/srt/test_expert_distribution.py
+++ b/test/srt/test_expert_distribution.py
@@ -8,7 +8,7 @@ import torch

 from sglang.srt.utils import kill_process_tree
 from sglang.test.test_utils import (
-    DEFAULT_SMALL_MOE_MODEL_NAME_FOR_TEST,
+    DEFAULT_SMALL_MOE_MODEL_NAME_FOR_TEST_BASE,
    DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
    DEFAULT_URL_FOR_TEST,
    CustomTestCase,