Migrate linter from pylint to ruff (#1665)

2023-11-20 11:58:01 -08:00
parent 112627e8b2
commit 5ffc0d13a2
45 changed files with 122 additions and 607 deletions
--- a/vllm/model_executor/models/chatglm.py
+++ b/vllm/model_executor/models/chatglm.py
@@ -269,10 +269,7 @@ class GLMTransformer(nn.Module):
        cache_events: Optional[List[torch.cuda.Event]],
    ) -> torch.Tensor:
        for i in range(self.num_layers):
-            if cache_events is None:
-                cache_event = None
-            else:
-                cache_event = cache_events[i]
+            cache_event = None if cache_events is None else cache_events[i]
            layer = self.layers[i]
            hidden_states = layer(
                hidden_states=hidden_states,