Fix 2-gpu CI test and suppress some warnings (#4930)

This commit is contained in:
Lianmin Zheng
2025-03-30 12:51:44 -07:00
committed by GitHub
parent f842853a40
commit 9adf178cc2
4 changed files with 21 additions and 19 deletions

View File

@@ -564,6 +564,10 @@ def load_image(image_file: Union[str, bytes]) -> tuple[Image, tuple[int, int]]:
def suppress_other_loggers():
warnings.filterwarnings(
"ignore", category=UserWarning, message="The given NumPy array is not writable"
)
try:
from vllm.logger import logger as vllm_default_logger
except ImportError:
@@ -578,10 +582,6 @@ def suppress_other_loggers():
)
logging.getLogger("vllm.config").setLevel(logging.ERROR)
warnings.filterwarnings(
"ignore", category=UserWarning, message="The given NumPy array is not writable"
)
def assert_pkg_version(pkg: str, min_version: str, message: str):
try: