code: def testNonScalarKeyUnorderedMap(self):
      with ops.Graph().as_default() as g:
          x = array_ops.placeholder(dtypes.float32)
          v = 2.0 * (array_ops.zeros([128, 128]) + x)
          t = data_flow_ops.gen_data_flow_ops.map_stage(
              key=constant_op.constant(
                  value=[1], shape=(1, 3), dtype=dtypes.int64
              ),
              indices=np.array([[6]]),
              values=[x, v],
              dtypes=[dtypes.int64],
              capacity=0,
              memory_limit=0,
              container="container1",
              shared_name="",
              name=None,
          )
  
      g.finalize()
  
      with self.session(graph=g) as sess:
          with self.assertRaisesRegex(
              errors.InvalidArgumentError, "key must be an int64 scalar"
          ):
              sess.run(t, feed_dict={x: 1})
  
imports: import queue
  import threading
  import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import data_flow_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 15 for calling the MapStageTest obj's underlying member method code with example usage: obj.testNonScalarKeyUnorderedMap() without return types