[CI][RFC] Replace black-jupyter with ruff-format (#37210)

Co-authored-by: Alison Shao <a.shao@wustl.edu>

python/sglang/srt/debug_utils/comparator/aligner/token_aligner/smart/aux_plugins.py

@@ -135,15 +135,15 @@ class _SGLangPlugin(_AuxFrameworkPlugin):
         seq_lens = step_data["seq_lens"]
         rids_raw = step_data.get("rids")
 
-        assert isinstance(
-            input_ids, torch.Tensor
-        ), f"input_ids: expected Tensor, got {type(input_ids)}"
-        assert isinstance(
-            positions, torch.Tensor
-        ), f"positions: expected Tensor, got {type(positions)}"
-        assert isinstance(
-            seq_lens, torch.Tensor
-        ), f"seq_lens: expected Tensor, got {type(seq_lens)}"
+        assert isinstance(input_ids, torch.Tensor), (
+            f"input_ids: expected Tensor, got {type(input_ids)}"
+        )
+        assert isinstance(positions, torch.Tensor), (
+            f"positions: expected Tensor, got {type(positions)}"
+        )
+        assert isinstance(seq_lens, torch.Tensor), (
+            f"seq_lens: expected Tensor, got {type(seq_lens)}"
+        )
 
         seq_lens_list: list[int] = seq_lens.tolist()
         num_seqs: int = len(seq_lens_list)

python/sglang/srt/debug_utils/comparator/aligner/token_aligner/smart/executor.py

@@ -108,7 +108,11 @@ def _build_bs_collapse_pattern(
 
     lhs: str = " ".join(names)  # type: ignore[arg-type]
 
-    rhs_names: list[str] = list(names[:lo]) + [f"({BATCH_DIM_NAME} {SEQ_DIM_NAME})"] + list(names[hi + 1 :])  # type: ignore[misc]
+    rhs_names: list[str] = (
+        list(names[:lo])
+        + [f"({BATCH_DIM_NAME} {SEQ_DIM_NAME})"]
+        + list(names[hi + 1 :])
+    )  # type: ignore[misc]
     rhs: str = " ".join(rhs_names)
 
     new_names: list[str | None] = (

python/sglang/srt/debug_utils/comparator/output_formatter.py

@@ -283,10 +283,7 @@ def _format_non_tensor_rich_body(
     target_val: str = escape(record.target_value)
 
     if record.values_equal:
-        return (
-            f"═ {name}{suffix} = {baseline_val} "
-            f"({record.baseline_type}) [green]✓[/]"
-        )
+        return f"═ {name}{suffix} = {baseline_val} ({record.baseline_type}) [green]✓[/]"
     return (
         f"═ [bold red]{name}{suffix}[/]\n"
         f"  baseline = {baseline_val} ({record.baseline_type})\n"

python/sglang/srt/debug_utils/cuda_coredump.py

@@ -86,7 +86,7 @@ def report():
     if not coredump_files:
         return
 
-    print(f"\n{'='*60}")
+    print(f"\n{'=' * 60}")
     print(f"CUDA coredump(s) detected ({len(coredump_files)} file(s)):")
     for f in coredump_files:
         size_mb = os.path.getsize(f) / (1024 * 1024)
@@ -98,7 +98,7 @@ def report():
         repo = os.environ.get("GITHUB_REPOSITORY", "sgl-project/sglang")
         print(f"Download from CI: gh run download {run_id} --repo {repo}")
 
-    print(f"{'='*60}\n")
+    print(f"{'=' * 60}\n")
 
 
 # Auto-inject CUDA coredump env vars at import time.

python/sglang/srt/debug_utils/dump_comparator.py

@@ -152,7 +152,7 @@ def check_tensor_pair(
         value_baseline = fn(x_baseline).item()
         value_target = fn(x_target).item()
         print(
-            f"[{name}] {value_baseline :.4f} vs {value_target:.4f} (diff: {value_target - value_baseline:.4f})"
+            f"[{name}] {value_baseline:.4f} vs {value_target:.4f} (diff: {value_target - value_baseline:.4f})"
         )
 
     if x_baseline.shape != x_target.shape:

python/sglang/srt/debug_utils/dump_loader.py

@@ -77,9 +77,9 @@ class DumpLoader:
         step = dumper._state.step
         conditions = dict(name=name, step=step, **kwargs)
         row = find_row(self._df, conditions=conditions)
-        assert (
-            row is not None
-        ), f"DumpLoader cannot find row given query {name=} {kwargs=} {self._directory=}"
+        assert row is not None, (
+            f"DumpLoader cannot find row given query {name=} {kwargs=} {self._directory=}"
+        )
 
         path = self._directory / row["filename"]
         output = torch.load(path, weights_only=False)

python/sglang/srt/debug_utils/dumper.py

@@ -179,9 +179,9 @@ class DumperConfig(_BaseConfig):
                 f"grafter_role must be 'baseline' or 'target' when grafter_enable=True, "
                 f"got {self.grafter_role!r}"
             )
-            assert (
-                self.grafter_master_address
-            ), "grafter_master_address must be set when grafter_enable=True"
+            assert self.grafter_master_address, (
+                "grafter_master_address must be set when grafter_enable=True"
+            )
             assert self.grafter_master_port > 0, (
                 f"grafter_master_port must be a positive port when grafter_enable=True, "
                 f"got {self.grafter_master_port}"
@@ -996,9 +996,9 @@ class _Grafter:
             return
 
         cfg = self._config
-        assert (
-            dist.is_initialized()
-        ), "[Grafter] default torch.distributed must be initialized"
+        assert dist.is_initialized(), (
+            "[Grafter] default torch.distributed must be initialized"
+        )
         role = _GraftRole(cfg.grafter_role)
         local_world = dist.get_world_size()
         local_rank = dist.get_rank()
@@ -1795,7 +1795,6 @@ class _SGLangPlugin(_FrameworkPlugin):
             return None
 
         try:
-
             args = get_server_args()
             if args is None:
                 return None

python/sglang/srt/debug_utils/tensor_dump_forward_hook.py

@@ -158,7 +158,7 @@ def register_forward_hook_for_model(
     model_top_level_module_matched, _ = tensor_dumper._add_hook_recursive(
         model, "", top_level_module_name, layers_module_name
     )
-    assert (
-        model_top_level_module_matched
-    ), f"model should have a module named {top_level_module_name}"
+    assert model_top_level_module_matched, (
+        f"model should have a module named {top_level_module_name}"
+    )
     return tensor_dumper