code: def load_mnist_model_softmax():
      """
      Model with a softmax layer last in the list of modules. Used to test the logic of pytorch_model.py,
      method get_softmax_arg_model (see the method's documentation).
      """
      model = torch.nn.Sequential(
          LeNet(),
          torch.nn.Softmax(),
      )
      return model
  
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 21 for calling a global function with example usage: load_mnist_model_softmax() and returns: model