code: def np_shape_array():
      flags = (
          mx.npx.is_np_shape(),
          mx.npx.is_np_array(),
          mx.npx.is_np_default_dtype(),
      )
      mx.npx.set_np()
      yield
      mx.npx.set_np(*flags)
  
imports: from contextlib import contextmanager
  import ctypes
  import numpy
  import pytest
  import mxnet
  from mxnet import amp
  from mxnet.base import check_call, _LIB
  from mxnet.gluon import nn
  from mxnet.test_utils import assert_allclose
label_desc: Use image node_id 1 for calling a global function with example usage: np_shape_array() without return types