"""GLVQ example using the MNIST dataset."""

from prototorch.models import ImageGLVQ
from pytorch_lightning.utilities.cli import LightningCLI

from mnist import TrainOnMNIST


class GLVQMNIST(TrainOnMNIST, ImageGLVQ):
    """Model Definition."""


cli = LightningCLI(GLVQMNIST)