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/opt.py
+++ b/vllm/model_executor/models/opt.py
@@ -257,10 +257,7 @@ class OPTDecoder(nn.Module):
        hidden_states = inputs_embeds + pos_embeds

        for i in range(len(self.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, kv_caches[i], input_metadata,
                                  cache_event)