From 0c87478713b3db7583da0ae950587e3316291004 Mon Sep 17 00:00:00 2001
From: Diego Montes <54745152+d57montes@users.noreply.github.com>
Date: Tue, 28 Sep 2021 15:22:31 -0400
Subject: [PATCH] Update wandb_utils.py (#4953)

`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
---
 utils/loggers/wandb/wandb_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py
index 9a80dc42ca95..92fdd27bb004 100644
--- a/utils/loggers/wandb/wandb_utils.py
+++ b/utils/loggers/wandb/wandb_utils.py
@@ -45,7 +45,8 @@ def check_wandb_config_file(data_config_file):
 
 
 def check_wandb_dataset(data_file):
-    is_wandb_artifact = False
+    is_trainset_wandb_artifact = False
+    is_valset_wandb_artifact = False
     if check_file(data_file) and data_file.endswith('.yaml'):
         with open(data_file, errors='ignore') as f:
             data_dict = yaml.safe_load(f)