def METHOD_NAME(num_classes: int = 10, class_type: str = "single") -> MnistNet:
    """
    MnistNet implementation; expected input shape is (B, 1, 28, 28)
    :param num_classes: the number of classes to classify
    :param class_type: one of [single, multi] to support multi class training;
        default single
    :return: The created MnistNet Module
    """
    return MnistNet(num_classes, class_type)