From 6575aea12804a5779e7502c35d188562ed3bc0ad Mon Sep 17 00:00:00 2001 From: Kangyan-Zhou Date: Wed, 29 Apr 2026 14:59:17 -0700 Subject: [PATCH] [CI] Fix black formatting on main (unblocks PR #21247 lint) (#24093) Co-authored-by: Claude Opus 4.7 (1M context) --- python/sglang/srt/model_executor/model_runner.py | 4 +--- python/sglang/srt/model_loader/loader.py | 7 ++----- python/sglang/srt/server_args.py | 5 ++++- 3 files changed, 7 insertions(+), 9 deletions(-) diff --git a/python/sglang/srt/model_executor/model_runner.py b/python/sglang/srt/model_executor/model_runner.py index 2b261fa83..57486df63 100644 --- a/python/sglang/srt/model_executor/model_runner.py +++ b/python/sglang/srt/model_executor/model_runner.py @@ -1003,9 +1003,7 @@ class ModelRunner(ModelRunnerKVCacheMixin): sv = torch.empty(0, dtype=torch.uint8, device=t.device).set_( t.untyped_storage() ) - if sv.data_ptr() not in { - v.data_ptr() for v in model_tensors.values() - }: + if sv.data_ptr() not in {v.data_ptr() for v in model_tensors.values()}: model_tensors[f"{name}.__storage"] = sv nixl_metadata = nixl_mgr.register_tensors(model_tensors) diff --git a/python/sglang/srt/model_loader/loader.py b/python/sglang/srt/model_loader/loader.py index 6b4450899..94ad0e4fe 100644 --- a/python/sglang/srt/model_loader/loader.py +++ b/python/sglang/srt/model_loader/loader.py @@ -2408,9 +2408,7 @@ class RemoteInstanceModelLoader(BaseModelLoader): # --- Transport-specific transfer --- if transport == "nixl": - self._transfer_via_nixl( - model, nixl_mgr, source_worker, tp_rank - ) + self._transfer_via_nixl(model, nixl_mgr, source_worker, tp_rank) else: self._transfer_via_transfer_engine( model, transfer_engine, source_worker, tp_rank @@ -2555,8 +2553,7 @@ class RemoteInstanceModelLoader(BaseModelLoader): ) logger.info( - "ModelExpress [nixl]: transferred %d tensors, " - "%.2f GB in %.2fs", + "ModelExpress [nixl]: transferred %d tensors, " "%.2f GB in %.2fs", matched, total_bytes / 1e9, duration, diff --git a/python/sglang/srt/server_args.py b/python/sglang/srt/server_args.py index b4edf2ad6..25242d665 100644 --- a/python/sglang/srt/server_args.py +++ b/python/sglang/srt/server_args.py @@ -7109,7 +7109,10 @@ class ServerArgs: if self.remote_instance_weight_loader_start_seed_via_transfer_engine: return True # ModelExpress source mode needs TransferEngine init only if transport is transfer_engine. - if self.modelexpress_source and self.modelexpress_transport == "transfer_engine": + if ( + self.modelexpress_source + and self.modelexpress_transport == "transfer_engine" + ): return True # Use TransferEngine as client backend. elif (