pymc-devs/pymc

Replace `assert` checks in `logprob.scan` with real error handling

Aberta

#7.780 aberto em 9 de mai. de 2025

 (1 comentário) (0 reação) (0 responsável)Python (1.902 forks)batch import
enhancementshackathonhelp wantedneeds infopytensor

Métricas do repositório

Stars
 (7.926 estrelas)
Métricas de merge de PR
 (Mesclagem média 11d) (7 fundiu PRs em 30d)

Description

Description

There are two annoying assert checks that I commonly run up against when writing time series models here and here. Other users are hitting them too, see for example here.

It would be much nicer to have a real error message that explains what has gone wrong and why, or at least gives some information about which parts of the graph are going wrong.

I'm not 100% sure how one ends up hitting these asserts, so I tagged this with needs info. It seems to have to do with having multiple outputs from the scan inner function in a custom dist.

Guia do colaborador