diff --git a/python/sglang/multimodal_gen/test/run_suite.py b/python/sglang/multimodal_gen/test/run_suite.py index fc5224774..046eb6005 100644 --- a/python/sglang/multimodal_gen/test/run_suite.py +++ b/python/sglang/multimodal_gen/test/run_suite.py @@ -29,14 +29,18 @@ _UPDATE_WEIGHTS_MODEL_PAIR_IDS = ( ) SUITES = { + # no GPU required; safe to run on any CPU-only runner + "unit": [ + "../unit/test_sampling_params_validate.py", + "../unit/test_storage.py", + "../unit/test_lora_format_adapter.py", + # add new unit tests here + ], "1-gpu": [ "test_server_a.py", "test_server_b.py", - "test_lora_format_adapter.py", # cli test "../cli/test_generate_t2i_perf.py", - # unit tests (no server needed) - "../test_sampling_params_validate.py", "test_update_weights_from_disk.py", # add new 1-gpu test files here ], diff --git a/python/sglang/multimodal_gen/test/server/test_lora_format_adapter.py b/python/sglang/multimodal_gen/test/unit/test_lora_format_adapter.py similarity index 100% rename from python/sglang/multimodal_gen/test/server/test_lora_format_adapter.py rename to python/sglang/multimodal_gen/test/unit/test_lora_format_adapter.py diff --git a/python/sglang/multimodal_gen/test/test_sampling_params_validate.py b/python/sglang/multimodal_gen/test/unit/test_sampling_params_validate.py similarity index 100% rename from python/sglang/multimodal_gen/test/test_sampling_params_validate.py rename to python/sglang/multimodal_gen/test/unit/test_sampling_params_validate.py diff --git a/python/sglang/multimodal_gen/test/test_storage.py b/python/sglang/multimodal_gen/test/unit/test_storage.py similarity index 100% rename from python/sglang/multimodal_gen/test/test_storage.py rename to python/sglang/multimodal_gen/test/unit/test_storage.py