moving QuadratureRule over to an equinox.Module and patched up a few …

[cmhamel]

…tests that was using the old QuadratureRule.len method. Now using a len method in QuadratureRule so we can use the python built-in len method.

[cmhamel]

@btalamini You suggested splitting up PR #87 since @ralberd was seeing some performance regressions from that initial stab I took which likely changed too much too fast.

This PR does something simple. Add equinox and jaxtyping as dependencies and converts the QuadratureRule namedtuple implementation to a equinox.Module. jaxtyping just helps document stuff better. It serves no functional purpose other than that.

@ralberd I ran that hole array problem you sent me ~10 times for each implementation and so no real difference.

Also, I left the current constructors in place so it didn't touch every file under the sun. With the equinox class though, you can directly call QuadratureRule(xiguass, wguass).

[codecov]

Codecov Report

Attention: Patch coverage is 90.62500% with 3 lines in your changes missing coverage. Please review.

Project coverage is 76.42%. Comparing base (9ea926d) to head (802ba44).
Report is 3 commits behind head on main.

Files with missing lines	Patch %	Lines
optimism/Mechanics.py	57.14%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #96      +/-   ##
==========================================
+ Coverage   76.40%   76.42%   +0.02%     
==========================================
  Files          61       61              
  Lines        5051     5056       +5     
==========================================
+ Hits         3859     3864       +5     
  Misses       1192     1192              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[btalamini]

This looks really useful! Thanks, Craig!

[cmhamel]

@ralberd @btalamini @tupek2

If it's ok with y'all, I can tag the main branch prior to merging this as say "v0.0.1" or "pre-equinox" so if we need to easily revert to pre-equinox stuff it should be fairly straightforward

[btalamini]

@ralberd @btalamini @tupek2

If it's ok with y'all, I can tag the main branch prior to merging this as say "v0.0.1" or "pre-equinox" so if we need to easily revert to pre-equinox stuff it should be fairly straightforward

That's a good suggestion. Go for it.