def METHOD_NAME(cls):
    # Get MNIST
    (x_train, y_train), (x_test, y_test), _, _ = load_dataset("mnist")
    x_train, y_train = x_train[:NB_TRAIN], y_train[:NB_TRAIN]
    x_test, y_test = x_test[:NB_TEST], y_test[:NB_TEST]
    cls.mnist = (x_train, y_train), (x_test, y_test)