Skip to content

Commit

Permalink
Merge pull request #9 from aimet-tech/update-wandb-utils
Browse files Browse the repository at this point in the history
determine entity and project when calling use_artifact
  • Loading branch information
aimet-pasitpk authored Mar 19, 2024
2 parents 563c098 + 5c87aba commit 2be6539
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aimet_ml/utils/wandb_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,4 +80,4 @@ def load_artifact(
if artifact_name_with_alias not in available_artifact_names:
return None

return wandb.use_artifact(artifact_name_with_alias)
return wandb.use_artifact(f"{api.settings['entity']}/{api.settings['project']}/{artifact_name_with_alias}")

0 comments on commit 2be6539

Please sign in to comment.