Update wandb_utils.py (#4953)

`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
This commit is contained in:
Diego Montes authored and GitHub committed 2021-09-28 12:22:31 -07:00
1 parent 38c779b099
commit 0c87478713
1 file changed
+2 -1
+2 -1
View File
@@ -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)