def METHOD_NAME(  # type: ignore
    self, step: Union[Step, StepInfo]
) -> Optional[wandb.apis.public.Artifact]:
    artifact_kind = (step.metadata or {}).get("artifact_kind", ArtifactKind.STEP_RESULT.value)
    try:
        return self.wandb_client.artifact(
            f"{self.entity}/{self.project}/{self._step_artifact_name(step)}:{step.unique_id}",
            type=artifact_kind,
        )
    except WandbError as exc:
        if is_missing_artifact_error(exc):
            return None
        else:
            raise