medrag_multi_modal/utils.py

import wandb


def get_wandb_artifact(artifact_name: str, artifact_type: str) -> str:
    if wandb.run:
        artifact = wandb.use_artifact(artifact_name, type=artifact_type)
        artifact_dir = artifact.download()
    else:
        api = wandb.Api()
        artifact = api.artifact(artifact_name)
        artifact_dir = artifact.download()
    return artifact_dir