dev: Check shapes in integrator.

[daniel-dodd]

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!

[github-actions]

There has been no recent activity on this issue. To keep our issues log clean, we remove old and inactive issues.
Please update to the latest version of GPJax and check if that resolves the issue. Let us know if that works for you by leaving a comment.
This issue is now marked as stale and will be closed if no further activity occurs. If you believe that this is incorrect, please comment. Thank you!

[github-actions]

There has been no activity on this PR for some time. Therefore, we will be automatically closing the PR if no new activity occurs within the next seven days.
Thank you for your contributions.