code: def make_regression_dataset(datatype, nrows, ncols, n_info):
      X, y = make_regression(
          n_samples=nrows,
          n_features=ncols,
          n_informative=5,
          random_state=0,
      )
      X = X.astype(datatype)
      y = y.astype(datatype)
  
      return X, y
  
file_path: cuml/python/cuml/tests/dask/test_dask_linear_regression.py
imports: from cuml.internals.safe_imports import gpu_only_import
  import pytest
  from cuml.dask.common import utils
  from sklearn.metrics import mean_squared_error
  from sklearn.datasets import make_regression
  from cuml.internals.safe_imports import cpu_only_import
label_desc: Use image node_id 2 for calling a global function with example usage: make_regression_dataset(datatype, nrows, ncols, n_info) and returns: X,  y