python - 使用 Kedro 挂钩将 git_sha 作为参数记录在 Mlflow 上

Question

我想在 Mlflow 上记录 git_sha 参数，如文档中所示。在我看来，只需运行以下代码部分就足以让 git_sha 登录到 Mlflow UI。我对吗 ？

@hook_impl
    def before_pipeline_run(self, run_params: Dict[str, Any]) -> None:
        """Hook implementation to start an MLflow run
        with the same run_id as the Kedro pipeline run.
        """
        mlflow.start_run(run_name=run_params["run_id"])
        mlflow.log_params(run_params)

但这不起作用，因为我得到了除了 git_sha 参数之外的所有参数。当我查看hooks specs时，似乎这个参数不是 run_params 的一部分（不再了吗？）

有没有办法可以获取 git sha（可能来自上下文日志？）并将其添加到记录的参数中？

先感谢您 ！

Answer 1

虽然强烈鼓励将 git 与 Kedro 一起使用，但这不是必需的，因此 Kedro 的任何部分（如果我们是迂腐的，除了kedro-starters之外）没有“意识到”git。

在您before_pipeline_hook那里，您很容易通过此处记录的技术检索信息。对于整个代码库而言，这似乎微不足道，如果您想说提供管道特定的哈希值，则涉及更多。