Merge pull request #551 from ankitgv0/patch-2

Logger bug fix
QData · Oct 22, 2021 · 1918bfd · 1918bfd
2 parents 1811105 + 8a3e8c2
commit 1918bfd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/textattack/trainer.py b/textattack/trainer.py
@@ -15,6 +15,7 @@
 import transformers
 
 import textattack
+from textattack.shared.utils import logger
 
 from .attack import Attack
 from .attack_args import AttackArgs
@@ -25,8 +26,6 @@
 from .models.wrappers import ModelWrapper
 from .training_args import CommandLineTrainingArgs, TrainingArgs
 
-logger = textattack.shared.utils.logger
-
 
 class Trainer:
     """Trainer is training and eval loop for adversarial training.