Skip to content

Commit

Permalink
Mypy type ignore
Browse files Browse the repository at this point in the history
Signed-off-by: Flavia Beo <[email protected]>
  • Loading branch information
flaviabeo committed Oct 29, 2024
1 parent 0739409 commit 44390a1
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions vllm/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,

Check failure on line 437 in vllm/config.py

View workflow job for this annotation

GitHub Actions / mypy (3.8)

Argument "pooling_type" to "PoolingConfig" has incompatible type "Optional[PoolingType]"; expected "PoolingType" [arg-type]

Check failure on line 437 in vllm/config.py

View workflow job for this annotation

GitHub Actions / mypy (3.9)

Argument "pooling_type" to "PoolingConfig" has incompatible type "Optional[PoolingType]"; expected "PoolingType" [arg-type]

Check failure on line 437 in vllm/config.py

View workflow job for this annotation

GitHub Actions / mypy (3.10)

Argument "pooling_type" to "PoolingConfig" has incompatible type "Optional[PoolingType]"; expected "PoolingType" [arg-type]

Check failure on line 437 in vllm/config.py

View workflow job for this annotation

GitHub Actions / mypy (3.11)

Argument "pooling_type" to "PoolingConfig" has incompatible type "Optional[PoolingType]"; expected "PoolingType" [arg-type]

Check failure on line 437 in vllm/config.py

View workflow job for this annotation

GitHub Actions / mypy (3.12)

Argument "pooling_type" to "PoolingConfig" has incompatible type "Optional[PoolingType]"; expected "PoolingType" [arg-type]
normalize=normalize) # type: ignore
if pooling_type_arg is not None:
pooling_config.pooling_type = self.get_pooling_type(
pooling_type_arg)
Expand Down

0 comments on commit 44390a1

Please sign in to comment.