super-gradients/src/super_gradients/training/exceptions/dataset_exceptions.py

class DatasetValidationException(Exception):
    pass


class ParameterMismatchException(DatasetValidationException):
    pass


class IllegalDatasetParameterException(DatasetValidationException):
    """
    Exception raised illegal dataset param.

    Attributes:
        message -- explanation of the error
    """

    def __init__(self, desc):
        self.message = "Unsupported dataset parameter format: " + desc
        super().__init__(self.message)


class EmptyDatasetException(DatasetValidationException):
    """
    Exception raised when a dataset does not have any image for a specific config

    Attributes:
        message -- explanation of the error
    """

    def __init__(self, desc):
        self.message = "Empty Dataset: " + desc
        super().__init__(self.message)


class UnsupportedBatchItemsFormat(ValueError):
    """Exception raised illegal batch items returned from data loader.

    Attributes:
        message -- explanation of the error
    """

    def __init__(self):
        self.message = (
            "Batch items returned by the data loader expected format: \n"
            "1. torch.Tensor or tuple, s.t inputs = batch_items[0], targets = batch_items[1] and len("
            "batch_items) = 2 \n"
            "2. tuple: (inputs, targets, additional_batch_items)"
        )
        super().__init__(self.message)