diff --git a/frame_semantic_transformer/data/loaders/framenet17/ensure_wordnet_downloaded.py b/frame_semantic_transformer/data/loaders/framenet17/ensure_wordnet_downloaded.py index 98ad527..664eb70 100644 --- a/frame_semantic_transformer/data/loaders/framenet17/ensure_wordnet_downloaded.py +++ b/frame_semantic_transformer/data/loaders/framenet17/ensure_wordnet_downloaded.py @@ -6,3 +6,7 @@ def ensure_wordnet_downloaded() -> None: nltk.data.find("corpora/wordnet.zip") except LookupError: nltk.download("wordnet") + try: + nltk.data.find("corpora/omw-1.4.zip") + except LookupError: + nltk.download("omw-1.4")