Início / coding / Perguntas / 79250860
Accepted
Gaussian97
Asked: 2024-12-04 18:56:45 +0800 CST

Dica de tipo para itertools.product não sabe o comprimento dos elementos

  • 772

Estou adicionando dicas de tipo a um código antigo meu. No entanto, encontro um "problema" que não sei como resolver. Tenho uma função que se parece com:

def f(x: tuple[tuple[int, int], ...]):
    ...

Ou seja, ele aceita um argumento que é uma tupla de pares int. Em algum ponto do código eu chamo essa função passando como argumento a saída de itertools.product

from itertools import product

x = tuple(product(range(10), repeat=2))
f(x)

O código funciona sem problemas, pois xé uma tupla de 100 pares inteiros. No entanto, mypy reclama com o erro

error: Argument 1 to "tuple" has incompatible type "product[tuple[int, ...]]"; expected "Iterable[tuple[int, int]]"  [arg-type]

Existe uma maneira de superar esse problema?

python

1 respostas

  1. Best Answer

    Isso acontece devido à seguinte sobrecarga ( detypeshed ):

    class product(Generic[_T_co]):
    ...
    @overload
    def __new__(cls, *iterables: Iterable[_T1], repeat: int = 1) -> product[tuple[_T1, ...]]: ...

    Devido às limitações do sistema de tipos, a única maneira de obter melhores resultados é ter algumas sobrecargas mais específicas no mesmo lugar:

    (parques infantis: Mypy , Pyright )

    @overload
def __new__(cls, iterable: Iterable[_T1], /, *, repeat: Literal[1] = 1) -> product[tuple[_T1]]: ...
@overload
def __new__(cls, iterable: Iterable[_T1], /, *, repeat: Literal[2]) -> product[tuple[_T1, _T1]]: ...
@overload
def __new__(cls, iterable: Iterable[_T1], /, *, repeat: Literal[3]) -> product[tuple[_T1, _T1, _T1]]: ...
# And so on
@overload
def __new__(cls, iterable: Iterable[_T1], /, *, repeat: int) -> product[tuple[_T1, ...]]: ...

    reveal_type(tuple(product(range(10), repeat=2)))  # tuple[tuple[int, int], ...]

    No entanto, não há tais sobrecargas. Nem é preciso dizer que essa abordagem não escala muito bem, já que o número de sobrecargas necessárias crescerá exponencialmente conforme o número de iterables aumenta, embora você possa tentar abrir um issue para solicitar isso.

    Enquanto isso, é necessário usar cast()ou # type: ignore:

    f(cast(Iterable[tuple[int, int]], x)
f(x)  # type: ignore
    • 1

relate perguntas