code: def create_plot(data, x_axis, y_axis, output_filepath=None):
      fig, ax = plt.subplots(figsize=(12, 3))
      im = ax.imshow(
          data, aspect="auto", origin="lower", interpolation="none"
      )
      plt.colorbar(im, ax=ax)
      plt.xlabel(x_axis)
      plt.ylabel(y_axis)
      plt.tight_layout()
  
      if output_filepath:
          plt.savefig(output_filepath, format="png")
  
      fig.canvas.draw()
      data = save_figure_to_numpy(fig)
      plt.close()
      return data
  
imports: from enum import Enum
  from typing import Optional, Tuple
  import librosa
  import matplotlib.pylab
  import numpy
  import torch
  from einops import rearrange
  from numba import jit, prange
  from nemo.collections.tts.torch.tts_data_types import DATA_STR2DATA_CLASS, MAIN_DATA_TYPES, WithLens
  from nemo.utils import logging
  from nemo.utils.decorators import deprecated
label_desc: Use image node_id 19 for calling a global function with example usage: create_plot(data, x_axis, y_axis, output_filepath) and returns: data