Update wandb_utils.py (#4953)

`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
pull/4979/head
Diego Montes 2021-09-28 15:22:31 -04:00 committed by GitHub
parent 38c779b099
commit 0c87478713
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

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)