[CI] Move tests onto the right CI stages (#34074)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>

test/registered/debug_utils/comparator/aligner/entrypoint/test_executor.py

@@ -33,8 +33,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=7, suite="base-c-test-cpu")
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 class TestExecuteSubPlans:

test/registered/debug_utils/comparator/aligner/entrypoint/test_planner.py

@@ -25,8 +25,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 def _make_meta(

test/registered/debug_utils/comparator/aligner/reorderer/test_executor.py

@@ -26,8 +26,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=6, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _zigzag_order(cp_size: int) -> list[int]:

test/registered/debug_utils/comparator/aligner/reorderer/test_planner.py

@@ -26,8 +26,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=7, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 class TestComputeReordererPlans:

test/registered/debug_utils/comparator/aligner/test_axis_aligner.py

@@ -17,8 +17,7 @@ from sglang.srt.debug_utils.comparator.log_sink import log_sink
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 class TestComputeAxisAlignerPlan:

test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_loader.py

@@ -19,7 +19,7 @@ from sglang.srt.debug_utils.comparator.log_sink import LogSink
 from sglang.srt.debug_utils.comparator.output_types import ErrorLog, InfoLog
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 _sglang_plugin = _SGLangPlugin()
 _megatron_plugin = _MegatronPlugin()

test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_plugins.py

@@ -16,8 +16,7 @@ from sglang.srt.debug_utils.comparator.aligner.token_aligner.smart.types import
 from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=6, suite="base-c-test-cpu")
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 _sglang_plugin = _SGLangPlugin()
 _megatron_plugin = _MegatronPlugin()

test/registered/debug_utils/comparator/aligner/token_aligner/test_concat_steps.py

@@ -10,8 +10,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import apply_dim_names
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=6, suite="base-c-test-cpu")
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 class TestExecuteConcat:

test/registered/debug_utils/comparator/aligner/token_aligner/test_executor.py

@@ -29,8 +29,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 def _named(tensor: torch.Tensor, names: list[str]) -> torch.Tensor:

test/registered/debug_utils/comparator/aligner/token_aligner/test_planner.py

@@ -23,8 +23,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=6, suite="base-c-test-cpu")
+register_cpu_ci(est_time=30, stage="weekly", runner_config="cpu")
 
 
 class TestBuildTokenIndexSGLangThd:

test/registered/debug_utils/comparator/aligner/token_aligner/test_thd_seq_lens_loader.py

@@ -14,7 +14,7 @@ from sglang.srt.debug_utils.comparator.aligner.token_aligner.smart.aux_plugins i
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 def _save_pt(

test/registered/debug_utils/comparator/aligner/unsharder/test_executor.py

@@ -30,8 +30,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 from sglang.srt.debug_utils.comparator.output_types import ReplicatedCheckResult
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _name_tensors(

test/registered/debug_utils/comparator/aligner/unsharder/test_parallel_info.py

@@ -9,8 +9,7 @@ from sglang.srt.debug_utils.comparator.aligner.unsharder.types import AxisInfo
 from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=6, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 class TestNormalizeParallelInfo:

test/registered/debug_utils/comparator/aligner/unsharder/test_planner.py

@@ -19,8 +19,7 @@ from sglang.srt.debug_utils.comparator.aligner.unsharder.types import (
 from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis, parse_dims
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 class TestComputeUnsharderPlan:

test/registered/debug_utils/comparator/dims_spec/test_dim_parser.py

@@ -12,8 +12,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=6, suite="base-c-test-cpu")
+register_cpu_ci(est_time=6, stage="weekly", runner_config="cpu")
 
 
 class TestParseDim:

test/registered/debug_utils/comparator/dims_spec/test_dims_parser.py

@@ -15,8 +15,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=8, stage="weekly", runner_config="cpu")
 
 
 class TestSingletonDimUtilFilterOut:

test/registered/debug_utils/comparator/dims_spec/test_tensor_naming.py

@@ -14,8 +14,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=7, suite="base-c-test-cpu")
+register_cpu_ci(est_time=7, stage="weekly", runner_config="cpu")
 
 
 class TestFindDimIndex:

test/registered/debug_utils/comparator/dims_spec/test_types.py

@@ -9,8 +9,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=8, stage="weekly", runner_config="cpu")
 
 
 class TestDimConstants:

test/registered/debug_utils/comparator/tensor_comparator/test_comparator.py

@@ -16,8 +16,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import DiffInfo
 from sglang.srt.debug_utils.comparator.threshold_dsl import DiffThresholdRule
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=20, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=7, suite="base-c-test-cpu")
+register_cpu_ci(est_time=20, stage="weekly", runner_config="cpu")
 
 
 class TestComputeTensorInfo:

test/registered/debug_utils/comparator/tensor_comparator/test_formatter.py

@@ -63,7 +63,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 _DEFAULT_PERCENTILE_LINES: list[str] = [
     "      [blue]p1        [/]    -1.8000      -1.8000   [dim]+0.00e+00[/]",

test/registered/debug_utils/comparator/tensor_comparator/test_types.py

@@ -21,8 +21,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _make_stats(**overrides) -> TensorStats:

test/registered/debug_utils/comparator/test_bundle_comparator.py

@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.srt.debug_utils.dump_loader import ValueWithMeta
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 def _save_tensor(

test/registered/debug_utils/comparator/test_bundle_matcher.py

@@ -13,7 +13,7 @@ from sglang.srt.debug_utils.comparator.bundle_matcher import (
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 def _make_row(

test/registered/debug_utils/comparator/test_display.py

@@ -21,7 +21,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _render_rich(renderable: object) -> str:

test/registered/debug_utils/comparator/test_dp_utils.py

@@ -12,7 +12,7 @@ from sglang.srt.debug_utils.comparator.dp_utils import (
 from sglang.srt.debug_utils.dump_loader import ValueWithMeta
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=15, stage="weekly", runner_config="cpu")
 
 
 def _make_sglang_meta(

test/registered/debug_utils/comparator/test_dump_loader.py

@@ -7,7 +7,7 @@ import torch
 from sglang.srt.debug_utils.dump_loader import read_tokenizer_path
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _save_pt(

test/registered/debug_utils/comparator/test_e2e_demo.py

@@ -26,7 +26,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
 from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="default", nightly=True)
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 _EXP_NAME = "demo_exp"
 

test/registered/debug_utils/comparator/test_entrypoint.py

@@ -30,7 +30,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
 from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper, _RecomputeStatus
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=30, stage="weekly", runner_config="cpu")
 
 _FIXED_EXP_NAME = "my_exp_name"
 

test/registered/debug_utils/comparator/test_log_sink.py

@@ -11,8 +11,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
 from sglang.srt.debug_utils.comparator.report_sink import report_sink
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=7, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _make_error_log(**overrides) -> ErrorLog:

test/registered/debug_utils/comparator/test_manually_verify.py

@@ -23,7 +23,7 @@ import torch
 
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=60, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=60, stage="weekly", runner_config="cpu")
 
 _PUBLISH_DIR: Path = Path("/tmp/comparator_manual_verify")
 _PNG_MAGIC: bytes = b"\x89PNG"

test/registered/debug_utils/comparator/test_meta_overrider.py

@@ -16,8 +16,7 @@ from sglang.srt.debug_utils.comparator.meta_overrider import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 # ───────────────────── Unit: MetaOverrideRule ─────────────────────

test/registered/debug_utils/comparator/test_model_validation.py

@@ -44,8 +44,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
 from sglang.srt.debug_utils.comparator.utils import Pair, _check_equal_lengths
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=7, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 class TestCheckEqualLengths:

test/registered/debug_utils/comparator/test_output_types.py

@@ -60,7 +60,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
 from sglang.srt.debug_utils.comparator.utils import Pair
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _render_rich(renderable: object) -> str:

test/registered/debug_utils/comparator/test_per_token_visualizer.py

@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.comparator import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=30, stage="weekly", runner_config="cpu")
 
 _PNG_MAGIC: bytes = b"\x89PNG"
 

test/registered/debug_utils/comparator/test_preset.py

@@ -3,8 +3,7 @@ import pytest
 from sglang.srt.debug_utils.comparator.preset import PRESETS, expand_preset
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=8, suite="base-c-test-cpu")
+register_cpu_ci(est_time=8, stage="weekly", runner_config="cpu")
 
 
 class TestExpandPreset:

test/registered/debug_utils/comparator/test_threshold_dsl.py

@@ -11,7 +11,7 @@ from sglang.srt.debug_utils.comparator.threshold_dsl import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 def _ev(

test/registered/debug_utils/comparator/test_utils.py

@@ -17,8 +17,7 @@ from sglang.srt.debug_utils.comparator.utils import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=7, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 class TestCalcRelDiff:

test/registered/debug_utils/comparator/test_visualizer.py

@@ -10,7 +10,7 @@ from sglang.srt.debug_utils.comparator.visualizer.preprocessing import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=30, stage="weekly", runner_config="cpu")
 
 
 class TestPreprocessTensor:

test/registered/debug_utils/comparator/testing_helpers.py

@@ -12,8 +12,8 @@ from sglang.test.ci.ci_register import register_cpu_ci
 
 register_cpu_ci(
     est_time=0,
-    suite="base-a-test-cpu",
-    nightly=True,
+    stage="weekly",
+    runner_config="cpu",
     disabled="helper module, no tests",
 )
 

test/registered/debug_utils/source_patcher/test_code_patcher.py

@@ -10,8 +10,7 @@ from sglang.srt.debug_utils.source_patcher.code_patcher import (
 from sglang.srt.debug_utils.source_patcher.types import EditSpec, PatchSpec
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=5, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
 

test/registered/debug_utils/source_patcher/test_dumper_integration.py

@@ -8,8 +8,7 @@ import yaml
 from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=6, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
 

test/registered/debug_utils/source_patcher/test_source_editor.py

@@ -5,8 +5,7 @@ from sglang.srt.debug_utils.source_patcher.source_editor import apply_edits
 from sglang.srt.debug_utils.source_patcher.types import EditSpec, PatchApplicationError
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=4, suite="base-c-test-cpu")
+register_cpu_ci(est_time=10, stage="weekly", runner_config="cpu")
 
 
 class TestApplyEdits:

test/registered/debug_utils/test_dump_comparator.py

@@ -9,7 +9,7 @@ from sglang.srt.debug_utils.dump_comparator import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=30, stage="weekly", runner_config="cpu")
 
 
 # ----------------------------- Unit tests -----------------------------

test/registered/debug_utils/test_dump_loader.py

@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.dump_loader import (
 )
 from sglang.test.ci.ci_register import register_cpu_ci
 
-register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
+register_cpu_ci(est_time=30, stage="weekly", runner_config="cpu")
 
 
 class TestReadMeta:

test/registered/debug_utils/test_schedule_simulator.py

@@ -25,8 +25,7 @@ from sglang.srt.debug_utils.schedule_simulator import (
 from sglang.test.ci.ci_register import register_cpu_ci
 from sglang.test.test_utils import CustomTestCase
 
-register_cpu_ci(est_time=120, suite="base-a-test-cpu", nightly=True)
-register_cpu_ci(est_time=357, suite="base-c-test-cpu")
+register_cpu_ci(est_time=357, stage="weekly", runner_config="cpu")
 
 # ==================== Non-E2E Tests ====================