mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-12 17:34:18 +00:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
|
import os
|
||
|
|
|
||
|
|
import datahub.emitter.mce_builder as builder
|
||
|
|
import datahub.metadata.schema_classes as models
|
||
|
|
from datahub.emitter.mcp import MetadataChangeProposalWrapper
|
||
|
|
from datahub.emitter.rest_emitter import DatahubRestEmitter
|
||
|
|
|
||
|
|
gms_server = os.getenv("DATAHUB_GMS_URL", "http://localhost:8080")
|
||
|
|
token = os.getenv("DATAHUB_GMS_TOKEN")
|
||
|
|
emitter = DatahubRestEmitter(gms_server=gms_server, token=token)
|
||
|
|
|
||
|
|
feature_urn = builder.make_ml_feature_urn(
|
||
|
|
feature_table_name="user_features",
|
||
|
|
feature_name="total_spend",
|
||
|
|
)
|
||
|
|
|
||
|
|
dataset_urn = builder.make_dataset_urn(
|
||
|
|
name="analytics.orders",
|
||
|
|
platform="snowflake",
|
||
|
|
env="PROD",
|
||
|
|
)
|
||
|
|
|
||
|
|
metadata_change_proposal = MetadataChangeProposalWrapper(
|
||
|
|
entityUrn=feature_urn,
|
||
|
|
aspect=models.MLFeaturePropertiesClass(
|
||
|
|
description="Total amount spent by user across all orders. "
|
||
|
|
"Version 2.0 now includes refunds and returns, providing net spend instead of gross. "
|
||
|
|
"Changed from gross spend calculation in v1.0.",
|
||
|
|
dataType="CONTINUOUS",
|
||
|
|
version=models.VersionTagClass(versionTag="2.0"),
|
||
|
|
sources=[dataset_urn],
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
emitter.emit(metadata_change_proposal)
|