1
0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00

add install of stanza resource when package is present

This commit is contained in:
Jin Yong Yoo
2020-10-31 00:49:53 -04:00
parent 4f16bb097b
commit e2dd75001c

View File

@@ -116,6 +116,13 @@ def _post_install():
nltk.download("wordnet")
nltk.download("punkt")
try:
import stanza
stanza.download("en")
except Exception:
pass
def set_cache_dir(cache_dir):
"""Sets all relevant cache directories to ``TA_CACHE_DIR``."""