or-toledano
commited on
Fix `ROOT / data` when running W&B `log_dataset()` (#6606)
Browse files* Fix missing data folder when running log_dataset
* Use ROOT/'data'
* PEP8 whitespace
utils/loggers/wandb/wandb_utils.py
CHANGED
@@ -356,7 +356,7 @@ class WandbLogger():
|
|
356 |
# create a _wandb.yaml file with artifacts links if both train and test set are logged
|
357 |
if not log_val_only:
|
358 |
path = (path.stem if overwrite_config else path.stem + '_wandb') + '.yaml' # updated data.yaml path
|
359 |
-
path =
|
360 |
data.pop('download', None)
|
361 |
data.pop('path', None)
|
362 |
with open(path, 'w') as f:
|
|
|
356 |
# create a _wandb.yaml file with artifacts links if both train and test set are logged
|
357 |
if not log_val_only:
|
358 |
path = (path.stem if overwrite_config else path.stem + '_wandb') + '.yaml' # updated data.yaml path
|
359 |
+
path = ROOT / 'data' / path
|
360 |
data.pop('download', None)
|
361 |
data.pop('path', None)
|
362 |
with open(path, 'w') as f:
|