from idiomify.fetchers import fetch_epie


def main():
    idioms = set([
        idiom
        for idiom, _, _ in fetch_epie()
    ])
    contexts = [
        context
        for _, _, context in fetch_epie()
    ]
    print("Total number of idioms:", len(idioms))
    # This should learn... this - what I need for now is building a datamodule out of this
    print("Total number of contexts:", len(contexts))


if __name__ == '__main__':
    main()