Improve compatibility with dask-expr

[hoxbro]

Fixes #1331

• Needs to add a test
• Add comment about the line

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 85.61%. Comparing base (7c1111c) to head (d9ba50a).
Report is 5 commits behind head on main.

❗ Current head d9ba50a differs from pull request most recent head 54f5453

Please upload reports for the commit 54f5453 to get more accurate results.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1335      +/-   ##
==========================================
+ Coverage   85.58%   85.61%   +0.03%     
==========================================
  Files          51       50       -1     
  Lines       11290    11263      -27     
==========================================
- Hits         9662     9643      -19     
+ Misses       1628     1620       -8