pymc-devs/pymc

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

Open

#7780 aperta il 9 mag 2025

Vedi su GitHub
 (1 commento) (0 reazioni) (0 assegnatari)Python (1902 fork)batch import
enhancementshackathonhelp wantedneeds infopytensor

Metriche repository

Star
 (7926 star)
Metriche merge PR
 (Merge medio 11g 6h) (12 PR mergiate in 30 g)

Descrizione

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.

Guida contributor