GPJax
GPJax copied to clipboard
dev: Check shapes in integrator.
Thankfully beartype told me there was an issue with my code during refactoring to CoLA #370, but we need to be really careful with shapes in the integrator for quadrature.
If we pass the wrong shapes through, we get the wrong answer.
We should include a shape check here!