Merge branch 'main' into add_typing_to_tests

* main:
  Try cleaning up some expected_groups logic (xarray-contrib#175)
  Update visualizations & docs (xarray-contrib#189)
  Avoid broadcasting by variables against each other (xarray-contrib#186)
  Some typing (xarray-contrib#190)

Author: dcherian

asv_bench/benchmarks/cohorts.py

@@ -93,7 +93,7 @@ def setup(self, *args, **kwargs):
         ret = flox.core._factorize_multiple(
             by,
             expected_groups=(pd.Index(np.arange(1, 13)), pd.Index(np.arange(1, 25))),
-            by_is_dask=False,
+            any_by_dask=False,
             reindex=False,
         )
         # Add one so the rechunk code is simpler and makes sense