def METHOD_NAME(self):
    """
    Test exporting a classification model to ONNX and running inference
    """
    path = "google/bert_uncased_L-2_H-128_A-2"
    trainer = HFTrainer()
    model, tokenizer = trainer(path, self.data)
    # Output file path
    output = os.path.join(tempfile.gettempdir(), "onnx")
    # Export model to ONNX
    onnx = HFOnnx()
    model = onnx((model, tokenizer), "text-classification", output, True)
    # Test classification
    labels = Labels((model, path), dynamic=False)
    self.assertEqual(labels("cat")[0][0], 1)