only import wandb when called upon

QData · May 14, 2020 · 28bab4e · 28bab4e
1 parent f781666
commit 28bab4e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/textattack/loggers/weights_and_biases_logger.py b/textattack/loggers/weights_and_biases_logger.py
@@ -1,10 +1,9 @@
-import wandb
-
 from textattack.shared.utils import html_table_from_rows
 from .logger import Logger
 
 class WeightsAndBiasesLogger(Logger):
     def __init__(self, filename='', stdout=False):
+        import wandb
         wandb.init(project='textattack')
         self._result_table_rows = []