Fix ROOT / data when running W&B log_dataset() (#6606)

* Fix missing data folder when running log_dataset

* Use ROOT/'data'

* PEP8 whitespace
This commit is contained in:
or-toledano authored and GitHub committed 2022-02-10 16:17:47 +01:00
1 parent c21da596f3
commit cb2ad9f685
1 file changed
+1 -1
+1 -1
View File
@@ -356,7 +356,7 @@ class WandbLogger():
# 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: