code: def test__get_target_from_target_fn__with_2d_output_argmax():
      preds = good_model(X_train).argmax(axis=1)
      target = _get_target_from_target_fn(
          np_argmax_fn, good_model, X_train
      )  # TODO: fix up types, though mypy doesn't complain
      assert_allclose(target, preds)
  
imports: from functools import partial
  import numpy
  import pytest
  import tensorflow
  from numpy.testing import assert_allclose
  from tensorflow.keras import Model
  from alibi.api.interfaces import Explanation
  from alibi.explainers import IntegratedGradients
  from alibi.explainers.integrated_gradients import _get_target_from_target_fn, _run_forward_from_layer, _run_forward_to_layer, _check_target, _select_target
label_desc: Use image node_id 25 for calling a global function with example usage: test__get_target_from_target_fn__with_2d_output_argmax() without return types