docs: def write_array(path, arr):
      """Write an array to a .npy file."""
      file_ext = op.splitext(path)[1]
      if file_ext == '.npy':
          return np.save(path, arr)
      raise NotImplementedError("The file extension `{}` ".format(file_ext) +
                                "is not currently supported.")