Extend CatchAuxError to non TensorTree aux

normal-computing / posteriors

Uncertainty quantification with PyTorch

https://normal-computing.github.io/posteriors/

Apache License 2.0

282 stars 12 forks source link

Closed SamDuffield closed 2 months ago

SamDuffield commented 2 months ago

Extend CatchAuxError scope to also throw an error when the auxiliary output is not a TensorTree
Fix tests