diff --git a/OpenAttack/attackers/__init__.py b/OpenAttack/attackers/__init__.py index 8bc264e..1cee8d1 100644 --- a/OpenAttack/attackers/__init__.py +++ b/OpenAttack/attackers/__init__.py @@ -17,4 +17,4 @@ from .pso import PSOAttacker from .bert_attack import BERTAttacker from .bae import BAEAttacker -from .geometry import GEOAttacker +# from .geometry import GEOAttacker FIXME: cannot import name 'zero_gradients' from 'torch.autograd.gradcheck' diff --git a/requirements-dev.txt b/requirements-dev.txt index fe5c3ce..1c1d5b6 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -3,7 +3,7 @@ twine language_tool_python nltk>=3.5 numpy -tqdm<4.50.0,>=4.27 datasets +tqdm transformers>=4.0.0 torch>=1.5.1 \ No newline at end of file diff --git a/requirements-doc.txt b/requirements-doc.txt index a7b06c8..84430e1 100644 --- a/requirements-doc.txt +++ b/requirements-doc.txt @@ -1,7 +1,7 @@ nltk numpy -tqdm<4.50.0,>=4.27 datasets +tqdm transformers torch sphinx-thunlp-theme>=0.0.9 diff --git a/requirements.txt b/requirements.txt index d949c2b..31c0d82 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ nltk>=3.5 numpy -tqdm<4.50.0,>=4.27 datasets +tqdm transformers>=4.0.0 torch>=1.5.1 \ No newline at end of file