def _read_numpy_item(input_file):
  """Read the item found in a numpy save file.

  Args:
    input_file: Filename string or FileLike object.

  Returns:
    The first item.
  """
  if isinstance(input_file, str):
    with tf.io.gfile.GFile(input_file, 'rb') as fh:
      item = numpy.load(fh, allow_pickle=True).item()
  else:
    item = numpy.load(input_file, allow_pickle=True).item()
  return item
Read the item found in a numpy save file.

  Args:
    input_file: Filename string or FileLike object.

  Returns:
    The first item.
  
language/nql/nql/io.py