def load_corpi(mode, corpi=CORPI): if isinstance(mode, Trigram): tg = mode mode = tg.mode else: tg = Trigram(mode=mode) for base, gz, precooked in corpi: if precooked: tg.load_trigrams(cooked_corpi_path(base, mode, gz)) else: fn = raw_corpi_path(base, gz) tg.import_text(fn) return tg