code: def array_module(gpu):
      if gpu:
          if cp is None:
              raise ImportError(
                  "Execute on GPU requires for `cupy` library"
              )
          return cp
  
      return np
  
file_path: mars/mars/tensor/array_utils.py
imports: from collections import defaultdict
  from contextlib import contextmanager
  import numpy
  from ..lib import sparse
  from ..lib.sparse.core import issparse, get_dense_module
  from ..utils import lazy_import
label_desc: Use image node_id 4 for calling a global function with example usage: array_module(gpu) and returns: np, cp