Move profiler controls to SchedulerProfilerManager (#25614)

2026-05-18 18:33:17 +08:00
parent b0a511560c
commit a35690f070
4 changed files with 386 additions and 405 deletions
@@ -229,9 +229,11 @@ class TestProfileMergerIntegration(unittest.TestCase):
        self.assertIn("merge_profiles", sig.parameters)

        # Test SchedulerProfilerMixin
-        from sglang.srt.managers.scheduler_profiler_mixin import SchedulerProfilerMixin
+        from sglang.srt.managers.scheduler_components.profiler_manager import (
+            SchedulerProfilerManager,
+        )

-        sig = inspect.signature(SchedulerProfilerMixin._init_profile)
+        sig = inspect.signature(SchedulerProfilerManager._init_profile)
        self.assertIn("merge_profiles", sig.parameters)

        # Test CLI profiler