Mypy type ignore

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Oct 29, 2024 · 83eec19 · 83eec19
1 parent 0739409
commit 83eec19
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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, # type: ignore
+                normalize=normalize) 
             if pooling_type_arg is not None:
                 pooling_config.pooling_type = self.get_pooling_type(
                     pooling_type_arg)