code: def setUp(self):
      audio_path = (
          os.getcwd()
          + "/../src/test/resources/audio/json/audio_floats.json"
      )
      self.data = (
          SparkSessionForTest.spark.read.option(
              "inferSchema", value=True
          )
          .json(audio_path)
          .select(
              col("float_array")
              .cast("array<float>")
              .alias("audio_content")
          )
      )
  
      self.audio_assembler = (
          AudioAssembler()
          .setInputCol("audio_content")
          .setOutputCol("audio_assembler")
      )
  
      # Edit Manually
      model_path = "exported_onnx/openai/whisper-tiny"
      self.speech_to_text = (
          WhisperForCTC.loadSavedModel(
              model_path, SparkSessionForTest.spark
          )
          .setInputCols("audio_assembler")
          .setOutputCol("text")
      )
  
      self.pipeline = Pipeline(
          stages=[self.audio_assembler, self.speech_to_text]
      )
      self.model = self.pipeline.fit(self.data)
      self.model.write().overwrite().save(
          "./tmp_Whisper_pipeline_model"
      )
  
imports: import unittest
  import os
  import pytest
  from pyspark.sql.functions import col
  from sparknlp.annotator import *
  from sparknlp.base import *
  from test.util import SparkSessionForTest
label_desc: Use image node_id 1 for calling the WhisperTestSetUp obj's underlying member method code with example usage: obj.setUp() without return types