Fix 2-gpu CI test and suppress some warnings (#4930)
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user