From fa908a6a15b33d00e2d9b463a93df748de00f5ea Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Fri, 4 Jun 2021 23:27:15 -0600
Subject: [PATCH] Fix wandb import issue

---
 codes/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/codes/train.py b/codes/train.py
index 16b5b44c..6c55bb10 100644
--- a/codes/train.py
+++ b/codes/train.py
@@ -270,6 +270,7 @@ class Trainer:
                 for ek, ev in eval_dict.items():
                     self.tb_logger.add_scalar(ek, ev, self.current_step)
                 if opt['wandb']:
+                    import wandb
                     wandb.log(eval_dict)