Início / coding / Perguntas / 79322010
Accepted
Pierre-olivier Gendraud
Asked: 2025-01-02 01:06:46 +0800 CST

Como fazer com que o mypy verifique corretamente o tipo de uma função usando functools.partial?

  • 772

Estou tentando criar uma função que retorna um callable parcialmente aplicado, mas estou tendo problemas com a verificação de tipo mypy.

Aqui está minha primeira implementação:

Ajude-me a explicar minha pergunta para o stackoverflow. Ou seja, encontre um título e o corpo

este código:

from collections.abc import Callable
from functools import partial

def f(i: int, j: float, k: int) -> int:
    return i + int(j) + k

def g(a: float) -> Callable[[int, int], int]:
    return partial(f, j=a)

fun: Callable[[int, int], int] = g(3.0)
r: int = fun(4, 5)
print(r)

Foi verificado com sucesso pelo mypy

mas não pode correr

r: int = fun(4, 5) TypeError: f() obteve vários valores para o argumento 'j'

para resolver esse problema, chamo a função com argumento nomeado

from functools import partial

def f(i: int, j: float, k: int) -> int:
    return i + int(j) + k

def g(a: float) -> Callable[[int, int], int]:
    return partial(f, j=a)

fun: Callable[[int, int], int] = g(3.0)
# line 12 in my code (where the error message comes from)
r: int = fun(i=4, k=5)
print(r)

funciona bem agora

mas a verificação mypy falha

main.py:12: erro: Argumento de palavra-chave inesperado "i" [call-arg] main.py:12: erro: Argumento de palavra-chave inesperado "k" [call-arg] Foram encontrados 2 erros em 1 arquivo (1 arquivo de origem foi verificado)

Existe uma maneira de anotar esse código para que ele seja executado corretamente e passe na verificação de tipo do mypy? Eu tentei várias combinações de dicas de tipo, mas não encontrei uma solução que satisfaça tanto o comportamento de tempo de execução quanto a verificação de tipo estática.

Eu sei que existe essa solução, sem usar Partial

from collections.abc import Callable 


def f(i :int,j : float,k :int) ->int:
    return i+int(j)+k
    
def g(a :float) -> Callable[[int,int],int]:
    def ret(i,k):
        return f(i,a,k)
    return ret

fun :Callable[[int,int],int]= g(3.0)
r : int = fun(4,5)
print(r)

Mas eu realmente quero usar Callable porque estou trabalhando com funções com muitos parâmetros e é muito mais simples apenas dizer quais parâmetros são substituídos

python

2 respostas

  1. Best Answer

    Callablenão é expressivo o suficiente para poder descrever a assinatura da função que você está retornando. Em vez disso, você deve usar um Protocol.

    Em vez de:

    # All this says is that the function takes two ints as argument. It does not
# say what the name of the argument is. So you cannot use keyword arguments.
# Even if that is what is required in the case of arg `k`
FuncType = Callable[[int,int],int]

    Usar:

    from typing import Protocol

# Callable takes two int args called `i` and `k`. `i` can be passed as a normal
# arg or as a keyword arg, and `k` MUST be passed as a keyword arg.
class FuncType(Protocol):
    def __call__(self, i: int, *, k: int) -> int: ...

    Assim, a última parte do seu código se tornaria:

    class FuncType(Protocol):
    def __call__(self, i: int, *, k: int) -> int: ...

def g(a: float) -> FuncType:
    # NB. mypy does not check this cast, even with the --strict flag
    return partial(f, j=a) 

fun: FuncType = g(3.0)
r: int = fun(4, j=5)
# OR
r = fun(i=4, j=5)
# mypy is happy with either
print(r)
    • 3

  2. exatamente a mesma solução proposta por Dunes acabei de remover *int porque não entendo porque ele é usado

    from typing import Protocol 

from functools import partial

class FuncType(Protocol):
    def __call__(self, i: int, k: int) -> int: ...

def f(i: int, j: float, k: int) -> int:
    return i + int(j) + k

def g(a: float) -> FuncType:
    return partial(f, j=a)

fun: FuncType = g(3.0)
# line 12 in my code (where the error message comes from)
r: int = fun(i=4, k=5)
print(r)
    • 0

relate perguntas