code: def testNonScalarKeyMapPeek(self):
      with self.assertRaisesRegex(
          errors.InvalidArgumentError, "key must be an int64 scalar"
      ):
          v = data_flow_ops.gen_data_flow_ops.map_peek(
              key=constant_op.constant(
                  value=[1], shape=(1, 3), dtype=dtypes.int64
              ),
              indices=np.array([[6]]),
              dtypes=[dtypes.int64],
              capacity=0,
              memory_limit=0,
              container="container1",
              shared_name="",
              name=None,
          )
          self.evaluate(v)
  
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 16 for calling the MapStageTest obj's underlying member method code with example usage: obj.testNonScalarKeyMapPeek() without return types