fix lint

Author: aolemila

python/sgl_jax/srt/configs/kimi_linear.py

@@ -142,4 +142,4 @@ def linear_layer_ids(self):
 
     @property
     def full_attention_layer_ids(self):
-        return [i for i in range(self.num_hidden_layers) if not self.is_kda_layer(i)]
+        return [i for i in range(self.num_hidden_layers) if not self.is_kda_layer(i)]

python/sgl_jax/srt/hf_transformers_utils.py

@@ -19,7 +19,6 @@
 from transformers.models.auto.modeling_auto import MODEL_FOR_CAUSAL_LM_MAPPING_NAMES
 
 from sgl_jax.srt.configs.kimi_linear import KimiLinearConfig
-
 from sgl_jax.srt.managers.tiktoken_tokenizer import TiktokenTokenizer
 from sgl_jax.srt.utils.common_utils import is_remote_url, lru_cache_frozenset
 

python/sgl_jax/srt/models/kimi_linear.py

@@ -838,4 +838,4 @@ def _create_layer_mappings(
         return mappings
 
 
-EntryClass = KimiLinearForCausalLM
+EntryClass = KimiLinearForCausalLM

python/sgl_jax/srt/models/mimo_v2_flash.py

@@ -816,7 +816,7 @@ def __call__(
         else:
             output = self.logits_processor(hidden_states, self.model.embed_tokens, logits_metadata)
 
-        return output, {"token_to_kv_pool": layers_kv_fused},True, layers_topk_ids
+        return output, {"token_to_kv_pool": layers_kv_fused}, True, layers_topk_ids
 
 
 EntryClass = [MiMoV2FlashForCausalLM]

python/sgl_jax/srt/models/qwen2.py

@@ -500,7 +500,7 @@ def __call__(
             output = self.logits_processor(hidden_states, self.lm_head, logits_metadata)
         else:
             output = self.logits_processor(hidden_states, self.model.embed_tokens, logits_metadata)
-        return output, {"token_to_kv_pool": layers_kv_fused},layers_callback_flag, None
+        return output, {"token_to_kv_pool": layers_kv_fused}, layers_callback_flag, None
 
 
 EntryClass = Qwen2ForCausalLM

python/sgl_jax/srt/models/qwen3.py

@@ -589,7 +589,7 @@ def __call__(
                 aux_hidden_states=aux_hidden_states,
             )
 
-        return output, {"token_to_kv_pool": layers_kv_fused},layers_callback_flag, None
+        return output, {"token_to_kv_pool": layers_kv_fused}, layers_callback_flag, None
 
 
 EntryClass = Qwen3ForCausalLM