def get_instance()

in src/ab/plugins/db/sqlite.py [0:0]

• 19 lines of code
• 7 McCabe index (conditional complexity)

    def get_instance(db: Sqlite, config: dict):
        '''
        args:
            config: {
                'type': random,
                'count': max sample size
                }
        '''
        config = app.config.FORCE_SAMPLER or config or app.config.SAMPLER

        assert isinstance(config.get('count'), int) and config['count'] > 0, \
            'sampler.count must be positive interger, not string'

        if not hasattr(config, 'type'):
            return RandomSampler(db, config)

        if config['type'] == 'random':
            return RandomSampler(db, config)
        elif config['type'] == 'head':
            return HeadSampler(db, config)
        else:
            raise ValueError('unknown sampler type:', config['type'])