code: def test_export_to_onnx(self):
      model = WaveGlowModel(wcfg)
      model = model.cuda().half()
      typecheck.set_typecheck_enabled(enabled=False)
      with tempfile.TemporaryDirectory() as tmpdir, model.nemo_infer():
          tmp_file_name = os.path.join(tmpdir, "waveglow.onnx")
  
          n_mels = 80
          # Test export.
          inp = input_example(n_mels)
          inp1 = taco2wg(*inp)
          inp2 = inp1
          res1 = model.waveglow(*inp1)
          res2 = model.waveglow(*inp2)
          assert torch.allclose(res1, res2, rtol=0.01, atol=0.1)
          WaveGlowModel.forward_for_export = forward_wrapper
          model.export(
              tmp_file_name,
              input_example=inp,
              verbose=False,
              check_trace=False,
              do_constant_folding=True,
          )
  
file_path: NeMo/tests/collections/tts/test_waveglow.py
imports: import os
  import tempfile
  import pytest
  import torch
  from omegaconf import DictConfig
  from nemo.collections.tts.models import WaveGlowModel
  from nemo.core.classes import typecheck
label_desc: Use image node_id 1 for calling the TestWaveGlow obj's underlying member method code with example usage: obj.test_export_to_onnx() without return types