code: def init_kv(stype="default"):
      """init kv"""
      kv = mx.kv.create()
      # single
      kv.init(3, mx.nd.zeros(shape=shape, stype=stype))
      # list
      kv.init(keys, [mx.nd.zeros(shape=shape, stype=stype)] * len(keys))
      return kv
  
imports: import mxnet
  import numpy
  import unittest
  from mxnet.test_utils import rand_ndarray, assert_almost_equal
  from common import assertRaises
  from mxnet.base import py_str, MXNetError
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: init_kv(stype) and returns: kv