alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

Merge branch 'master' of github.com:QData/TextAttack into gpu-alloc

Browse Source
This commit is contained in:
Jack Morris
2020-08-18 12:25:28 -04:00
parent 37138a6183 579a81d7ae
commit 5c1d03bdfa
26 changed files with 370 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
textattack/shared/utils/install.py
View File

@@ -105,16 +105,15 @@ logger.propagate = False
def _post_install():
logger.info(
"First time running textattack: downloading remaining required packages."
)
logger.info("Updating TextAttack package dependencies.")
logger.info("Downloading NLTK required packages.")
import nltk
nltk.download("wordnet")
nltk.download("averaged_perceptron_tagger")
nltk.download("universal_tagset")
nltk.download("stopwords")
nltk.download("omw")
nltk.download("universal_tagset")
nltk.download("wordnet")
def set_cache_dir(cache_dir):
@@ -132,7 +131,7 @@ def set_cache_dir(cache_dir):
def _post_install_if_needed():
"""Runs _post_install if hasn't been run since install."""
# Check for post-install file.
post_install_file_path = path_in_cache("post_install_check")
post_install_file_path = path_in_cache("post_install_check_2")
post_install_file_lock_path = post_install_file_path + ".lock"
post_install_file_lock = filelock.FileLock(post_install_file_lock_path)
post_install_file_lock.acquire()