class TFStackGen: def __init__(self, mem, memKeys): self.tree = TreeStack() self.mem = mem self.memKeys = memKeys def gen_goals(self, verbose): world = self.mem.get(self.memKeys.MEM_STATES)[-1] blocks = blockstate.get_block_list(world) return [self.tree.givegoal(blocks)]
def give_goal(self, blockset): tree = TreeScen() return tree.givegoal(blockset)
def __init__(self, mem, memKeys): self.tree = TreeStack() self.mem = mem self.memKeys = memKeys