mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-11 17:03:33 +00:00
15 lines
464 B
Python
15 lines
464 B
Python
from datahub.metadata.urns import MlModelGroupUrn
|
|
from datahub.sdk import DataHubClient, GlossaryTermUrn
|
|
|
|
client = DataHubClient.from_env()
|
|
|
|
group_urn = MlModelGroupUrn(platform="mlflow", name="recommendation-models", env="PROD")
|
|
|
|
mlmodel_group = client.entities.get(group_urn)
|
|
|
|
mlmodel_group.add_term(GlossaryTermUrn("Recommendation"))
|
|
|
|
client.entities.update(mlmodel_group)
|
|
|
|
print(f"Added term {GlossaryTermUrn('Recommendation')} to ML model group {group_urn}")
|