Generalize handling of chunked array types

pyproject.toml

@@ -39,6 +39,7 @@ module = [
   "cf_units.*",
   "cfgrib.*",
   "cftime.*",
+  "cubed.*",
   "cupy.*",
   "fsspec.*",
   "h5netcdf.*",

xarray/core/dataarray.py

@@ -1265,6 +1265,7 @@ def chunk(
         token: str | None = None,
         lock: bool = False,
         inline_array: bool = False,
+        from_array_kwargs=None,
         **chunks_kwargs: Any,
     ) -> T_DataArray:
         """Coerce this array's data into a dask arrays with the given chunks.
@@ -1329,6 +1330,7 @@ def chunk(
             token=token,
             lock=lock,
             inline_array=inline_array,
+            from_array_kwargs=from_array_kwargs,
         )
         return self._from_temp_dataset(ds)
 

xarray/core/dataset.py

@@ -271,18 +271,25 @@ def _maybe_chunk(
     name_prefix="xarray-",
     overwrite_encoded_chunks=False,
     inline_array=False,
+    from_array_kwargs=None,
 ):
     from dask.base import tokenize
 
     if chunks is not None:
         chunks = {dim: chunks[dim] for dim in var.dims if dim in chunks}
     if var.ndim:
         # when rechunking by different amounts, make sure dask names change
-        # by provinding chunks as an input to tokenize.
+        # by providing chunks as an input to tokenize.
         # subtle bugs result otherwise. see GH3350
         token2 = tokenize(name, token if token else var._data, chunks)
         name2 = f"{name_prefix}{name}-{token2}"
-        var = var.chunk(chunks, name=name2, lock=lock, inline_array=inline_array)
+        var = var.chunk(
+            chunks,
+            name=name2,
+            lock=lock,
+            inline_array=inline_array,
+            from_array_kwargs=from_array_kwargs,
+        )
 
         if overwrite_encoded_chunks and var.chunks is not None:
             var.encoding["chunks"] = tuple(x[0] for x in var.chunks)
@@ -2199,6 +2206,7 @@ def chunk(
         token: str | None = None,
         lock: bool = False,
         inline_array: bool = False,
+        from_array_kwargs=None,
         **chunks_kwargs: None | int | str | tuple[int, ...],
     ) -> T_Dataset:
         """Coerce all arrays in this dataset into dask arrays with the given
@@ -2261,7 +2269,16 @@ def chunk(
             )
 
         variables = {
-            k: _maybe_chunk(k, v, chunks, token, lock, name_prefix)
+            k: _maybe_chunk(
+                k,
+                v,
+                chunks,
+                token,
+                lock,
+                name_prefix,
+                inline_array=inline_array,
+                from_array_kwargs=from_array_kwargs,
+            )
             for k, v in self.variables.items()
         }
         return self._replace(variables)