Update wandb_utils.py (ultralytics#4953)

`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
fhkiel-mlaip · Sep 28, 2021 · ad1fe7f · ad1fe7f
1 parent 9c41e1c
commit ad1fe7f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)