File size: 1,016 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import json
def add_wandb_visualization(run, mlpipeline_ui_metadata_path):
"""NOTE: To use this, you must modify your component to have an output called `mlpipeline_ui_metadata_path` AND call `wandb.init` yourself inside that component.
Example usage:
def my_component(..., mlpipeline_ui_metadata_path: OutputPath()):
import wandb
from wandb.integration.kfp.helpers import add_wandb_visualization
with wandb.init() as run:
add_wandb_visualization(run, mlpipeline_ui_metadata_path)
... # the rest of your code here
"""
def get_iframe_html(run):
return f'<iframe src="{run.url}?kfp=true" style="border:none;width:100%;height:100%;min-width:900px;min-height:600px;"></iframe>'
iframe_html = get_iframe_html(run)
metadata = {
"outputs": [{"type": "markdown", "storage": "inline", "source": iframe_html}]
}
with open(mlpipeline_ui_metadata_path, "w") as metadata_file:
json.dump(metadata, metadata_file)
|