mlflow-token

为部署在OAuth2-proxy和keycloak后面的MLFlow实例获取访问令牌。

此脚本将使用您的当前设置 MLFLOW_TRACKING_URI 来查找OAuth2-proxy的keycloak重定向。从那里它将启动OAuth设备流程，允许您获取有效的访问令牌。您可以使用此命令执行以下操作以更新您的 MLFLOW_TRACKING_TOKEN

% export $(mlflow-token)

并按照提示操作。

Jupyter Notebook中的使用

如果您想从Jupyter notebook内部对MLFlow实例进行认证，您可以将以下行添加到一个单元中

import mlflow_token
mlflow_token.setup_mlflow_environment("https://mlflow-demo.software-dev.ncsa.illinois.edu/")

这将更新笔记本的 os.environ，因此您可以立即使用mlflow SDK。令牌最终会过期，因此您可能需要偶尔重新运行单元格。

单元格将打印用户需要访问的url，并等待设备流程完成。