From 44390a12e6a732b1cb9a1673a10adc63839b8248 Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Tue, 29 Oct 2024 13:43:02 -0300 Subject: [PATCH] Mypy type ignore Signed-off-by: Flavia Beo --- vllm/config.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vllm/config.py b/vllm/config.py index 0540dc4170584..ed0192b1ca62b 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -433,8 +433,9 @@ def get_pooling_config( pooling_type = self.get_pooling_type( pooling_config["pooling_type"]) normalize = pooling_config["normalize"] - pooling_config = PoolingConfig(pooling_type=pooling_type, - normalize=normalize) + pooling_config = PoolingConfig( + pooling_type=pooling_type, + normalize=normalize) # type: ignore if pooling_type_arg is not None: pooling_config.pooling_type = self.get_pooling_type( pooling_type_arg)