`ProposalStateType`, `IntegratorStateType`, etc should be `namedtuple`s

aesara-devs / aehmc

An HMC/NUTS implementation in Aesara

MIT License

33 stars 6 forks source link

`ProposalStateType`, `IntegratorStateType`, etc should be `namedtuple`s #67

Closed rlouf closed 1 year ago

rlouf commented 2 years ago

Even if we can only pass tensors in functions used in scan this will make this code a lot easier to follow, and ready in case it becomes possible to pass namedtuples in scan.

__Originally posted by @brandonwillard in https://github.com/aesara-devs/aehmc/pull/7#discussion_r678733304__

zoj613 commented 1 year ago

What would be attributes of a TerminationState tuple? @rlouf