or-toledano commited on
Commit
cb2ad9f
·
unverified ·
1 Parent(s): c21da59

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

Files changed (1) hide show
  1. utils/loggers/wandb/wandb_utils.py +1 -1
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 = Path('data') / 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: