Skip to content

Commit

Permalink
Fix ROOT / data when running W&B log_dataset() (#6606)
Browse files Browse the repository at this point in the history
* Fix missing data folder when running log_dataset

* Use ROOT/'data'

* PEP8 whitespace
  • Loading branch information
or-toledano committed Feb 10, 2022
1 parent c21da59 commit cb2ad9f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/loggers/wandb/wandb_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit cb2ad9f

Please sign in to comment.