Format linting

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Oct 28, 2024 · ae73f4b · ae73f4b
1 parent 9c32660
commit ae73f4b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/vllm/config.py b/vllm/config.py
@@ -439,8 +439,7 @@ def get_pooling_config(
             if not pooling_type_arg and not normalize_arg:
                 return PoolingConfig(
                     pooling_type=self.get_pooling_type(pooling_type),
-                    normalize=normalize
-                )
+                    normalize=normalize)
         if pooling_type_arg:
             pooling_type = pooling_type_arg
         if normalize_arg:

diff --git a/vllm/transformers_utils/config.py b/vllm/transformers_utils/config.py
@@ -317,7 +317,7 @@ def get_pooling_config(model, revision='main', token: Optional[str] = None):
                                            token)
         pooling_type_name = next(
             (item for item, val in pooling_dict.items() if val is True), None)
-        
+
         pooling_type_name = get_pooling_config_name(pooling_type_name)
 
         return {"pooling_type": pooling_type_name, "normalize": normalize}
@@ -336,7 +336,7 @@ def get_pooling_config_name(pooling_name: str):
         pooling_name = "last"
 
     pooling_type_name = pooling_name.upper()
-    
+
     return pooling_type_name