Skip to content

Commit e93b6b9

Browse files
or-toledanoeladco
authored andcommitted
Fix ROOT / data when running W&B log_dataset() (ultralytics#6606)
* Fix missing data folder when running log_dataset * Use ROOT/'data' * PEP8 whitespace
1 parent 6a72dc6 commit e93b6b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/loggers/wandb/wandb_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ def log_dataset_artifact(self, data_file, single_cls, project, overwrite_config=
356356
# create a _wandb.yaml file with artifacts links if both train and test set are logged
357357
if not log_val_only:
358358
path = (path.stem if overwrite_config else path.stem + '_wandb') + '.yaml' # updated data.yaml path
359-
path = Path('data') / path
359+
path = ROOT / 'data' / path
360360
data.pop('download', None)
361361
data.pop('path', None)
362362
with open(path, 'w') as f:

0 commit comments

Comments
 (0)