code: def fix_get_mnist_subset(get_mnist_dataset):
      (x_train_mnist, y_train_mnist), (
          x_test_mnist,
          y_test_mnist,
      ) = get_mnist_dataset
      n_train = 50
      n_test = 50
      yield x_train_mnist[:n_train], y_train_mnist[
          :n_train
      ], x_test_mnist[:n_test], y_test_mnist[:n_test]
  
file_path: adversarial-robustness-toolbox/tests/defences/preprocessor/test_inverse_gan.py
imports: from __future__ import absolute_import, division, print_function, unicode_literals
  import pytest
  import logging
  import numpy
  from art.defences.preprocessor.inverse_gan import InverseGAN
  from art.attacks.evasion import FastGradientMethod
  from tests.utils import get_gan_inverse_gan_ft
  from tests.utils import ARTTestException
label_desc: Use image node_id 1 for calling a global function with example usage: fix_get_mnist_subset(get_mnist_dataset) without return types