code: def load_mnist_images_tf(load_mnist_images):
      """Load a batch of MNIST digits: inputs and outputs to use for testing."""
  
      return {
          "x_batch": np.moveaxis(load_mnist_images["x_batch"], 1, -1),
          "y_batch": load_mnist_images["y_batch"],
      }
  
imports: import pytest
  import pickle
  import torch
  import numpy
  from keras.datasets import cifar10
  import pandas
  from sklearn.model_selection import train_test_split
  import os
  from quantus.helpers.model.models import LeNet, LeNetTF, CifarCNNModel, ConvNet1D, ConvNet1DTF, TitanicSimpleTFModel, TitanicSimpleTorchModel
label_desc: Use image node_id 9 for calling a global function with example usage: load_mnist_images_tf(load_mnist_images) and returns: dict