Fix/quadratic pytree

[michalk8]

No description provided.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 89.40%. Comparing base (7cfd393) to head (892ac15).

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #555   +/-   ##
=======================================
  Coverage   89.39%   89.40%           
=======================================
  Files          71       71           
  Lines        7556     7543   -13     
  Branches     1080     1075    -5     
=======================================
- Hits         6755     6744   -11     
+ Misses        650      649    -1     
+ Partials      151      150    -1     

Files	Coverage Δ
src/ott/geometry/regularizers.py	99.54% <100.00%> (+0.83%)	⬆️