mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-14 18:31:59 +00:00
18 lines
490 B
Python
18 lines
490 B
Python
from datahub.sdk import DataHubClient
|
|
from datahub.sdk.mlmodelgroup import MLModelGroup
|
|
|
|
client = DataHubClient.from_env()
|
|
|
|
mlmodel_group = client.entities.get(
|
|
MLModelGroup.get_urn_type()(
|
|
platform="mlflow", name="recommendation-models", env="PROD"
|
|
)
|
|
)
|
|
|
|
doc_url = "https://wiki.example.com/ml/recommendation-models"
|
|
doc_description = "Model architecture and training documentation"
|
|
|
|
mlmodel_group.add_link((doc_url, doc_description))
|
|
|
|
client.entities.update(mlmodel_group)
|