jax-ml/jax

improve pjit pytree error involving Nones (and PytreeLeafs)

Aperta

#13.074 aperta il 2 nov 2022

Vedi su GitHubGuida contributor
 (6 commenti) (3 reazioni) (1 assegnatario)Python (3738 fork)github user discovery
better_errorsgood first issue

Metriche repository

Star
 (36.180 stelle)
Metriche merge PR
 (Merge medio 2g 19h) (439 PR mergiate in 30 g)

Descrizione

When pjit's in_axis_resoures is not a pytree prefix of the corresponding arguments' pytrees, we attempt to raise a good error. But at least two things can be improved:

  • don't mention PytreeLeaf instances in the error message
  • handle None specially (since it's surprising for the error message to refer to it as a container)

Guida contributor

Direzione di ricerca
Investiga la logica di validazione degli argomenti della funzione pjit, individua dove viene sollevato il messaggio di errore per struttura pytree non corrispondente e modifica la gestione degli errori per evitare di menzionare istanze PytreeLeaf e per trattare None come una foglia anziché come un contenitore.
Tech stack
python
Dominio
machine learningai
Tipo issue
Refactoring
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
GitPython
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
65