def METHOD_NAME(self) -> Optional[Iterable[Artifact]]:
    if mlflow_version < Version("2.0.0"):
        dir_path = self.mlflow_client.download_artifacts(
            run_id=self.run.info.run_id, path=""
        )
    else:
        dir_path = mlflow.METHOD_NAME.download_artifacts(run_id=self.run.info.run_id)
    artifact_name = self._handle_incompatible_strings(self.display_name())
    art = wandb.Artifact(artifact_name, "imported-artifacts")
    art.add_dir(dir_path)
    return [art]