Decorated __init__ confuses callback type if decorator is defined with type alias or if decorator is imported using relative import
Nessuno ha ancora preso questa issue.
- Lingua principale
- Python
- Stelle
- 20.6k
- Fork
- 3.3k
- Metriche di merge delle PR
- Metriche PR in attesa
Descrizione
This one's just weird. I'm not sure if this is a dup of #1927, but I figured I'd raise it anyway.
# test_case.py
from typing import Callable, Iterable, TypeVar, Union
_T = TypeVar("_T")
def _identity(__: _T) -> _T:
return __
# decr: Callable[[_T], _T] = _identity # <-- this approach works
DecoratorT = Callable[[_T], _T]
decr: DecoratorT = _identity # <-- this causes the false positives below
class Result:
@decr
def __init__(self):
pass
res = Result() # ; reveal_type(res) # -> Any (should be test_case.Result)
def some_results(op: Callable[[], Union[Result, Iterable[Result]]]) -> Iterable[Result]:
result_or_results = op()
results: Iterable[Result]
if isinstance(result_or_results, Result):
results = (result_or_results,) # ; reveal_type(result_or_results) # -> (should be test_case.Result)
else:
results = result_or_results # ; reveal_type(result_or_results) # -> Union[test_case.Result, typing.Iterable[test_case.Result]] (should be typing.Iterable[test_case.Result])
return results
% python --version ; mypy --version
Python 3.9.7
mypy 0.910
% mypy --config-file=pyproject.toml test_case.py
test_case.py:26: error: Incompatible types in assignment (expression has type "Union[Result, Iterable[Result]]", variable has type "Iterable[Result]")
Found 1 error in 1 file (checked 1 source file)
Note that if DecoratorT is defined as follows, things work again:
# …
class DecoratorT(Protocol):
def __call__(self, __: _T) -> _T:
...
decr: DecoratorT = _identity # <-- all good
# …
But if we move the decorator into its own module and use a relative import, it breaks no matter how we define the decorator type:
# test_case.py
from typing import Callable, Iterable, TypeVar, Union
# from test_case_import import decr # <-- this works
from .test_case_import import decr # <-- this causes the false positives below
class Result:
@decr
def __init__(self):
pass
res = Result() # ; reveal_type(res) # -> Any (should be test_case.Result)
def some_results(op: Callable[[], Union[Result, Iterable[Result]]]) -> Iterable[Result]:
result_or_results = op()
results: Iterable[Result]
if isinstance(result_or_results, Result):
results = (result_or_results,) # ; reveal_type(result_or_results) # -> (should be test_case.Result)
else:
results = result_or_results # ; reveal_type(result_or_results) # -> Union[test_case.Result, typing.Iterable[test_case.Result]] (should be typing.Iterable[test_case.Result])
return results
# test_case_import.py
from typing import Callable, TypeVar
_T = TypeVar("_T")
def _identity(__: _T) -> _T:
return __
decr: Callable[[_T], _T] = _identity # <-- doesn't matter how this is defined
% mypy --config-file=pyproject.toml test_case.py test_case_import.py
test_case.py:19: error: Incompatible types in assignment (expression has type "Union[Result, Iterable[Result]]", variable has type "Iterable[Result]")
Found 1 error in 1 file (checked 2 source files)
Using the Protocol approach does not salvage the second (relative import) scenario.
Guida per i contributori
Apri la guida per i contributori
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Direzione di ricerca
Inizia riproducendo le diagnosi segnalate con mypy in test_case.py e test_case_import.py, confrontando i casi del decorator locale e del decorator importato relativamente. Traccia come mypy inferisce il init decorato e i tipi del decorator importato. Il lavoro è completato quando gli esempi inferiscono Result e Iterable[Result] come annotato, senza il falso positivo relativo all'assegnazione.
Scritto dal modello di indicizzazione a partire dal testo della issue.
Valutazione
- Stack tecnologico
- python
- Ambito
- devtools
- Tipo di issue
- Bug
- Difficoltà
- 4/5
- Tempo stimato
- 3-5 giorni
- Stato di attività
- Ferma
- Chiarezza
- Abbastanza chiara
- Idoneità per principianti
- 35/100