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")
          )
      )
  
      audio_assembler = (
          AudioAssembler()
          .setInputCol("audio_content")
          .setOutputCol("audio_assembler")
      )
  
      speech_to_text = (
          Wav2Vec2ForCTC.pretrained()
          .setInputCols("audio_assembler")
          .setOutputCol("text")
      )
  
      pipeline = Pipeline(stages=[audio_assembler, speech_to_text])
      self.model = pipeline.fit(self.data)
      self.model.write().overwrite().save(
          "./tmp_wav2vec2_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 Wav2Vec2TestSetUp obj's underlying member method code with example usage: obj.setUp() without return types