Tiny fix for 1D arrays.

daskms/experimental/utils.py

@@ -138,6 +138,7 @@ def rechunk_by_size(dataset, max_chunk_mem=2**31 - 1, unchunked_dims=None):
     rechunked_dataset : xarray.Dataset
         Dataset with appropriate chunking.
     """
+
     def _rechunk(data_array, unchunked_dims):
         dims = set(data_array.dims)
         unchunked_dims = unchunked_dims & dims
@@ -162,6 +163,9 @@ def _rechunk(data_array, unchunked_dims):
 
         chunk_dict = {k: dim_sizes[k] for k in unchunked_dims}
 
+        if n_chunked_dim == 0:  # No chunking but still less than target size.
+            return chunk_dict
+
         ideal_chunk = int(
             np.power(max_n_ele / fixed_n_ele, 1 / (n_dim - n_unchunked_dim))
         )