def __init__(self, args, config): tmp = args.data_sampling args.data_sampling = 1 generator = Constructor(args, config) generator.initialize() args.data_sampling = tmp replacement_dict = config.lm.deterministic_replacement_dict_f( f=generator.all_f) self.searcher = TrieTree([]) for e in generator.expressions_list: if e.is_hyps != 2: t, _ = generator.encode_expr( e.tree, [generator.expressions_list[i].tree for i in e.hyps], replacement_dict) self.searcher.insert(t) return