modify TYPE_CHECKING

Signed-off-by: Shanshan Shen <[email protected]>
vllm-project · Dec 28, 2024 · 547c444 · 547c444
1 parent abf1f7e
commit 547c444
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/vllm/platforms/cuda.py b/vllm/platforms/cuda.py
@@ -15,14 +15,14 @@
 import vllm._C  # noqa
 import vllm.envs as envs
 from vllm.logger import init_logger
-from vllm.lora.punica_wrapper.punica_base import PunicaWrapperBase
 from vllm.lora.punica_wrapper.punica_gpu import PunicaWrapperGPU
 from vllm.utils import print_info_once
 
 from .interface import DeviceCapability, Platform, PlatformEnum
 
 if TYPE_CHECKING:
     from vllm.config import VllmConfig
+    from vllm.lora.punica_wrapper.punica_base import PunicaWrapperBase
 else:
     VllmConfig = None
 

diff --git a/vllm/platforms/hpu.py b/vllm/platforms/hpu.py
@@ -3,14 +3,14 @@
 import torch
 
 from vllm.logger import init_logger
-from vllm.lora.punica_wrapper.punica_base import PunicaWrapperBase
 from vllm.lora.punica_wrapper.punica_hpu import PunicaWrapperHPU
 from vllm.utils import print_info_once
 
 from .interface import Platform, PlatformEnum, _Backend
 
 if TYPE_CHECKING:
     from vllm.config import VllmConfig
+    from vllm.lora.punica_wrapper.punica_base import PunicaWrapperBase
 else:
     VllmConfig = None
 

diff --git a/vllm/platforms/rocm.py b/vllm/platforms/rocm.py
@@ -6,14 +6,14 @@
 
 import vllm.envs as envs
 from vllm.logger import init_logger
-from vllm.lora.punica_wrapper.punica_base import PunicaWrapperBase
 from vllm.lora.punica_wrapper.punica_gpu import PunicaWrapperGPU
 from vllm.utils import print_info_once
 
 from .interface import DeviceCapability, Platform, PlatformEnum, _Backend
 
 if TYPE_CHECKING:
     from vllm.config import VllmConfig
+    from vllm.lora.punica_wrapper.punica_base import PunicaWrapperBase
 else:
     VllmConfig = None