使用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规范时，这个参数似乎不再是run_params的一部分了(现在？(

有没有一种方法可以获取git-sha(可能来自上下文日志？(并将其添加到记录的参数中？

提前谢谢！

Answer 1

虽然我们强烈鼓励将git与Kedro一起使用，但这不是必需的，因此Kedro的任何部分(如果我们是迂腐的话，除了Kedro初学者(都不"知道"git。

在您的before_pipeline_hook中，您可以很容易地通过此处记录的技术检索信息。对于整个代码库来说，这似乎是微不足道的，如果你想说提供特定于管道的哈希，那就需要更多的参与。