python / python/mypy

Checking a match on an enum becomes quadratically slow after the variable was narrowed by an == check

Aperta
#21,997 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

performance topic-enum
Lingua principale
Python
Stelle
20.6k
Fork
3.3k
Metriche di merge delle PR
Metriche PR in attesa

Descrizione

Bug Report

Type-checking a match on a large enum becomes very slow when the variable was first narrowed by an == check with an early exit:

from enum import StrEnum

class E(StrEnum):
    M0 = "m0"
    M1 = "m1"
    # ... 1,200 members

def f(x: E) -> None:
    if x == E.M0:
        return
    match x:
        case E.M1:
            pass
        # ... through case E.M7

Removing the x == E.M0 block makes this much faster (~1.5s vs 22s locally).

To Reproduce

Your Environment

  • Mypy version used: 2.3.1, 2.4.0+dev.4d195a2
  • Python version used: 3.12

Guida per i contributori

Apri la guida per i contributori

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia con la riproduzione collegata su Playground e confronta il tempo del controllo dei tipi con e senza il controllo anticipato x == E.M0. Traccia i percorsi di restringimento degli enum e di verifica dei match in mypy, quindi verifica che l'esempio con un enum grande non diventi più lento in modo quadratico, preservando al contempo il comportamento di restringimento segnalato.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
python
Ambito
compilers, devtools
Tipo di issue
Bug
Difficoltà
4/5
Tempo stimato
3-5 giorni
Stato di attività
Attiva
Chiarezza
Abbastanza chiara
Idoneità per principianti
55/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.