File size: 259 Bytes
bfdae48 | 1 2 3 4 5 6 7 8 9 10 11 | _SAMPLERS = {}
def register_sampler(cls):
name = cls.__name__
if name in _SAMPLERS:
raise ValueError(f'{name} is already registered!')
_SAMPLERS[name] = cls
return cls
def get_sampler(cfg):
return _SAMPLERS[cfg.sampler.name](cfg) |