From cb2ad9f68531e6afe76326d46acf566acf8af4f9 Mon Sep 17 00:00:00 2001 From: or-toledano <49984106+or-toledano@users.noreply.github.com> Date: Thu, 10 Feb 2022 17:17:47 +0200 Subject: [PATCH] Fix `ROOT / data` when running W&B `log_dataset()` (#6606) * Fix missing data folder when running log_dataset * Use ROOT/'data' * PEP8 whitespace --- utils/loggers/wandb/wandb_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py index e81c33c54626..3c6b65723325 100644 --- a/utils/loggers/wandb/wandb_utils.py +++ b/utils/loggers/wandb/wandb_utils.py @@ -356,7 +356,7 @@ def log_dataset_artifact(self, data_file, single_cls, project, overwrite_config= # create a _wandb.yaml file with artifacts links if both train and test set are logged if not log_val_only: path = (path.stem if overwrite_config else path.stem + '_wandb') + '.yaml' # updated data.yaml path - path = Path('data') / path + path = ROOT / 'data' / path data.pop('download', None) data.pop('path', None) with open(path, 'w') as f: