From f2033e4b8faaab48929f4bca8a32e112efa364ca Mon Sep 17 00:00:00 2001 From: Judy Date: Thu, 23 May 2024 07:04:51 +0000 Subject: [PATCH] fix: fix import error #16 --- trainers/dcn_trainer.py | 2 ++ trainers/mf_trainer.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/trainers/dcn_trainer.py b/trainers/dcn_trainer.py index f0067c4..4819b6f 100644 --- 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 index f39da10..cf8f6a5 100644 --- 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