class_docstr: Tests tabular hot-flips on classification tasks.
class_docstr_tok: ["Tests","tabular","hot-flips","on","classification","tasks","."]
code: def test_mtc_prediction_is_argmax(self):
      output = self._gen.generate(
          example=self._example,
          model=self._model,
          dataset=self._dataset,
          config=self._config,
      )
      y_actual = output[0]["animal"]
      y_expected = self._predict_and_return_argmax_label(output[0])
      self.assertEqual(y_actual, y_expected)
  
imports: from unittest import mock
  from absl.testing import absltest
  from lit_nlp.api import dataset
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.components import minimal_targeted_counterfactuals
  from lit_nlp.lib import caching
  import numpy
  import scipy.special
label_desc: Use image node_id 9 for calling the ClassificationTabularMtcTest obj's underlying member method code with example usage: obj.test_mtc_prediction_is_argmax() without return types