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 = 10
      n_test = 10
      yield x_train_mnist[:n_train], y_train_mnist[
          :n_train
      ], x_test_mnist[:n_test], y_test_mnist[:n_test]
  
imports: import logging
  import numpy
  import pytest
  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