Browse Source

Explicitly convert artifact path to posix_path (#3067)

* Explicitly convert artifact path to posix_path

* Remove redudant str() casting
modifyDataloader
Ayush Chaurasia GitHub 3 years ago
parent
commit
d2a17289c9
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      utils/wandb_logging/wandb_utils.py

+ 2
- 1
utils/wandb_logging/wandb_utils.py View File

@@ -158,7 +158,8 @@ class WandbLogger():

def download_dataset_artifact(self, path, alias):
if isinstance(path, str) and path.startswith(WANDB_ARTIFACT_PREFIX):
dataset_artifact = wandb.use_artifact(remove_prefix(path, WANDB_ARTIFACT_PREFIX) + ":" + alias)
artifact_path = Path(remove_prefix(path, WANDB_ARTIFACT_PREFIX) + ":" + alias)
dataset_artifact = wandb.use_artifact(artifact_path.as_posix())
assert dataset_artifact is not None, "'Error: W&B dataset artifact doesn\'t exist'"
datadir = dataset_artifact.download()
return datadir, dataset_artifact

Loading…
Cancel
Save