from contextlib import suppress

HabanaAI · Sep 25, 2024 · 7ff33bf · 7ff33bf
1 parent e4f6d9d
commit 7ff33bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/worker/habana_model_runner.py b/vllm/worker/habana_model_runner.py
@@ -1941,8 +1941,8 @@ def execute_model(
 
     def shutdown_inc(self):
         can_finalize_inc = False
-        import contextlib
-        with contextlib.suppress(AttributeError):
+        from contextlib import suppress
+        with suppress(AttributeError):
             can_finalize_inc = (self.model_config.quantization == 'inc') and \
                 (self.model.model is not None) and \
                 self.inc_initialized_successfully and \