code: def init_kv_with_str(stype="default", kv_type="local"):
      """init kv"""
      kv = mx.kv.create(kv_type)
      # single
      kv.init("a", mx.nd.zeros(shape, stype=stype))
      # list
      kv.init(
          str_keys, [mx.nd.zeros(shape=shape, stype=stype)] * len(keys)
      )
      return kv
  
file_path: mxnet/tests/python/gpu/test_kvstore_gpu.py
imports: import sys
  import os
  import mxnet
  import numpy
  import pytest
  from mxnet.test_utils import assert_almost_equal, default_device, environment
label_desc: Use image node_id 1 for calling a global function with example usage: init_kv_with_str(stype, kv_type) and returns: kv