fix: fix import error #16

twndus · May 23, 2024 · f2033e4 · f2033e4
1 parent 6d9eaea
commit f2033e4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/trainers/dcn_trainer.py b/trainers/dcn_trainer.py
@@ -1,3 +1,5 @@
+import wandb
+
 import numpy as np
 import pandas as pd
 from tqdm import tqdm

diff --git a/trainers/mf_trainer.py b/trainers/mf_trainer.py
@@ -1,3 +1,5 @@
+import wandb
+
 import numpy as np
 import pandas as pd
 from tqdm import tqdm