code: def validate_dask_array(darray, client=None):
      if len(darray.chunks) > 2:
          raise ValueError(
              "Input array cannot have more than two dimensions"
          )
      elif len(darray.chunks) == 2 and len(darray.chunks[1]) > 1:
          raise ValueError("Input array cannot be chunked along axis 1")
  
file_path: cuml/python/cuml/dask/common/dask_arr_utils.py
imports: from cuml.common import rmm_cupy_ary, has_scipy
  from cuml.dask.common.part_utils import _extract_partitions
  from dask.distributed import default_client
  from cuml.dask.common.dask_df_utils import to_dask_cudf
  from cuml.internals.memory_utils import with_cupy_rmm
  import dask.dataframe
  import dask
  from cuml.internals.safe_imports import gpu_only_import
  from cuml.internals.safe_imports import cpu_only_import
label_desc: Use image node_id 1 for calling a global function with example usage: validate_dask_array(darray, client) without return types