polish type assert

Signed-off-by: youkaichao <[email protected]>
vllm-project · Jan 22, 2025 · 23ee3ad · 23ee3ad
1 parent daf3169
commit 23ee3ad
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vllm/device_allocator/cumem.py b/vllm/device_allocator/cumem.py
@@ -225,8 +225,9 @@ def use_memory_pool(self, tag: Optional[str] = None):
         """
         if tag is None:
             tag = CuMemAllocator.default_tag
-        else:
-            assert isinstance(tag, str)
+
+        assert isinstance(tag, str)
+
         old_tag = self.current_tag
         self.current_tag = tag
         with use_memory_pool_with_allocator(self.python_malloc_callback,