code: def _check_axis_partitioning(chunks, n_features):
      c = chunks[1][0]
      if c != n_features:
          msg = (
              "Can only generate arrays partitioned along the "
              "first axis. Specifying a larger chunksize for "
              "the second axis.\n\n\tchunk size: {}\n"
              "\tn_features: {}".format(c, n_features)
          )
          raise ValueError(msg)
  
file_path: dask-ml/dask_ml/datasets.py
imports: import numbers
  from datetime import timedelta
  import dask
  import dask.array
  import dask.dataframe
  import numpy
  import sklearn.datasets
  import sklearn.utils
  import dask_ml.utils
label_desc: Use image node_id 1 for calling a global function with example usage: _check_axis_partitioning(chunks, n_features) without return types