In PR ( #7162 ), we needed to add a .compute(scheduler="sync") call in one case (see below). While this seems ok for the cases where we needed it (concrete arrays, scalars, and Python objects). There's the potential this could be used on other objects (like Dask Arrays) where we would not want to use this. It would be good to figure out another way to solve this problem (if possible) that does not rely on the usage of compute.

cc @pentschev @jrbourbeau