Diego Montes commited on
Commit
0c87478
1 Parent(s): 38c779b

Update wandb_utils.py (#4953)

Browse files

`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.

Files changed (1) hide show
  1. utils/loggers/wandb/wandb_utils.py +2 -1
utils/loggers/wandb/wandb_utils.py CHANGED
@@ -45,7 +45,8 @@ def check_wandb_config_file(data_config_file):
45
 
46
 
47
  def check_wandb_dataset(data_file):
48
- is_wandb_artifact = False
 
49
  if check_file(data_file) and data_file.endswith('.yaml'):
50
  with open(data_file, errors='ignore') as f:
51
  data_dict = yaml.safe_load(f)
 
45
 
46
 
47
  def check_wandb_dataset(data_file):
48
+ is_trainset_wandb_artifact = False
49
+ is_valset_wandb_artifact = False
50
  if check_file(data_file) and data_file.endswith('.yaml'):
51
  with open(data_file, errors='ignore') as f:
52
  data_dict = yaml.safe_load(f)