fix: auto-download omw-1.4 for inference

chanind · Mar 15, 2023 · 343906c · 343906c
1 parent 73d1d44
commit 343906c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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")