fixed cupy failures in balanced_allocation and corridor#985

brendancol merged 1 commit intomasterxarray-contrib/xarray-spatial:masterfrom

cupy-test-fixesxarray-contrib/xarray-spatial:cupy-test-fixesCopy head branch name to clipboard