def set_gen(self, gen): from pymc.data import GeneratorAdapter if not isinstance(gen, GeneratorAdapter): gen = GeneratorAdapter(gen) if not gen.tensortype == self.generator.tensortype: raise ValueError("New generator should yield the same type") self.generator = gen
def __init__(self, gen, default=None): from pymc.data import GeneratorAdapter super().__init__() if not isinstance(gen, GeneratorAdapter): gen = GeneratorAdapter(gen) self.generator = gen self.set_default(default)