datahub/metadata-ingestion/examples/library/dataset_add_domain.py
2025-08-21 10:22:08 -07:00

15 lines
458 B
Python

from datahub.metadata.urns import DatasetUrn, DomainUrn
from datahub.sdk import DataHubClient
client = DataHubClient.from_env()
dataset = client.entities.get(DatasetUrn(platform="snowflake", name="example_dataset"))
# If you don't know the domain urn, you can look it up:
# domain_urn = client.resolve.domain(name="marketing")
# NOTE: This will overwrite the existing domain
dataset.set_domain(DomainUrn(id="marketing"))
client.entities.update(dataset)