code: def test_model_saves_with_example_input_array(
      tmpdir, modelclass, input_sample
  ):
      """Test that ONNX model saves with example_input_array and size is greater than 3 MB."""
      model = modelclass()
      model.example_input_array = input_sample
  
      file_path = os.path.join(tmpdir, "model.onnx")
      model.to_onnx(file_path)
      assert os.path.exists(file_path) is True
      assert os.path.getsize(file_path) > 4e2
  
file_path: pytorch-lightning/tests/tests_pytorch/models/test_onnx.py
imports: import operator
  import os
  from unittest.mock import patch
  import numpy
  import onnxruntime
  import pytest
  import torch
  from lightning.pytorch import Trainer
  from lightning.pytorch.demos.boring_classes import BoringModel
  from lightning_utilities import compare_version
  import tests_pytorch.helpers.pipelines
  from tests_pytorch.helpers.runif import RunIf
  from tests_pytorch.utilities.test_model_summary import UnorderedModel
label_desc: Use image node_id 3 for calling a global function with example usage: test_model_saves_with_example_input_array(tmpdir, modelclass, input_sample) without return types