def __init__()

in base/base_dataset.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, db_path, set_type, transform=None):
        """Init class

        Arguments:
            db_path {str} -- path to set
            set_type {SetType} -- set

        Keyword Arguments:
            transform {BaseTransform} -- transformation to apply to data (default: {None})
        """

        assert isinstance(set_type, SetType)
        self.logger = ConsoleLogger(set_type.value)

        if io.exists(db_path):
            self.path = db_path
        else:
            self.logger.error('Dataset directory {} not found'.format(db_path))

        self.index = self._load_index()

        if transform:
            assert isinstance(transform, (BaseTransform, transforms.Compose))
        self.transform = transform