code: def test_classification_zoo(arch_name):
      batch_size = 16
      # Model
      predictor = classification.zoo.crop_orientation_predictor(
          arch_name, pretrained=False
      )
      with pytest.raises(ValueError):
          predictor = classification.zoo.crop_orientation_predictor(
              arch="wrong_model", pretrained=False
          )
      # object check
      assert isinstance(predictor, CropOrientationPredictor)
      input_tensor = tf.random.uniform(
          shape=[batch_size, 128, 128, 3], minval=0, maxval=1
      )
      out = predictor(input_tensor)
      assert isinstance(out, list) and len(out) == batch_size
      assert all(isinstance(pred, int) for pred in out)
  
file_path: doctr/tests/tensorflow/test_models_classification_tf.py
imports: import os
  import tempfile
  import cv2
  import numpy
  import onnxruntime
  import psutil
  import pytest
  import tensorflow
  from doctr.models import classification
  from doctr.models.classification.predictor import CropOrientationPredictor
  from doctr.models.utils import export_model_to_onnx
label_desc: Use image node_id 3 for calling a global function with example usage: test_classification_zoo(arch_name) without return types