acryl-datahub 1.0.0rc8__py3-none-any.whl → 1.0.0rc10__py3-none-any.whl

datahub/metadata/_urns/urn_defs.py

@@ -10,7 +10,7 @@
 
 # This file contains classes corresponding to entity URNs.
 
-from typing import ClassVar, List, Optional, Type, TYPE_CHECKING, Union
+from typing import ClassVar, List, Optional, Type, TYPE_CHECKING, Union, Literal
 
 import functools
 from deprecated.sphinx import deprecated as _sphinx_deprecated
@@ -25,7 +25,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import BusinessAttributeKeyClass
 
 class BusinessAttributeUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "businessAttribute"
+    ENTITY_TYPE: ClassVar[Literal["businessAttribute"]] = "businessAttribute"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["BusinessAttributeUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -81,7 +81,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataTypeKeyClass
 
 class DataTypeUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataType"
+    ENTITY_TYPE: ClassVar[Literal["dataType"]] = "dataType"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataTypeUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -137,7 +137,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataProductKeyClass
 
 class DataProductUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataProduct"
+    ENTITY_TYPE: ClassVar[Literal["dataProduct"]] = "dataProduct"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataProductUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -193,7 +193,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import TelemetryKeyClass
 
 class TelemetryUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "telemetry"
+    ENTITY_TYPE: ClassVar[Literal["telemetry"]] = "telemetry"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, name: Union["TelemetryUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -249,7 +249,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubAccessTokenKeyClass
 
 class DataHubAccessTokenUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubAccessToken"
+    ENTITY_TYPE: ClassVar[Literal["dataHubAccessToken"]] = "dataHubAccessToken"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubAccessTokenUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -305,7 +305,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubConnectionKeyClass
 
 class DataHubConnectionUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubConnection"
+    ENTITY_TYPE: ClassVar[Literal["dataHubConnection"]] = "dataHubConnection"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubConnectionUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -361,7 +361,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubRoleKeyClass
 
 class DataHubRoleUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubRole"
+    ENTITY_TYPE: ClassVar[Literal["dataHubRole"]] = "dataHubRole"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubRoleUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -417,7 +417,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import MLModelKeyClass
 
 class MlModelUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "mlModel"
+    ENTITY_TYPE: ClassVar[Literal["mlModel"]] = "mlModel"
     _URN_PARTS: ClassVar[int] = 3
 
     def __init__(self, platform: Union["DataPlatformUrn", str], name: str, env: str = "PROD", *, _allow_coercion: bool = True) -> None:
@@ -480,7 +480,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import NotebookKeyClass
 
 class NotebookUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "notebook"
+    ENTITY_TYPE: ClassVar[Literal["notebook"]] = "notebook"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, notebook_tool: str, notebook_id: str, *, _allow_coercion: bool = True) -> None:
@@ -542,7 +542,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import RoleKeyClass
 
 class RoleUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "role"
+    ENTITY_TYPE: ClassVar[Literal["role"]] = "role"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["RoleUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -598,7 +598,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import GlobalSettingsKeyClass
 
 class GlobalSettingsUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "globalSettings"
+    ENTITY_TYPE: ClassVar[Literal["globalSettings"]] = "globalSettings"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["GlobalSettingsUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -654,7 +654,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DatasetKeyClass
 
 class DatasetUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataset"
+    ENTITY_TYPE: ClassVar[Literal["dataset"]] = "dataset"
     _URN_PARTS: ClassVar[int] = 3
 
     def __init__(self, platform: Union["DataPlatformUrn", str], name: str, env: str = "PROD", *, _allow_coercion: bool = True) -> None:
@@ -746,7 +746,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import ChartKeyClass
 
 class ChartUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "chart"
+    ENTITY_TYPE: ClassVar[Literal["chart"]] = "chart"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, dashboard_tool: str, chart_id: str, *, _allow_coercion: bool = True) -> None:
@@ -788,6 +788,18 @@ class ChartUrn(_SpecificUrn):
     def from_key_aspect(cls, key_aspect: "ChartKeyClass") -> "ChartUrn":
         return cls(dashboard_tool=key_aspect.dashboardTool, chart_id=key_aspect.chartId)
 
+    @classmethod
+    def create_from_ids(
+        cls,
+        platform: str,
+        name: str,
+        platform_instance: Optional[str] = None,
+    ) -> "ChartUrn":
+        return ChartUrn(
+            dashboard_tool=platform,
+            chart_id=f"{platform_instance}.{name}" if platform_instance else name,
+        )
+        
     @property
     def dashboard_tool(self) -> str:
         return self._entity_ids[0]
@@ -800,7 +812,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import GlossaryNodeKeyClass
 
 class GlossaryNodeUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "glossaryNode"
+    ENTITY_TYPE: ClassVar[Literal["glossaryNode"]] = "glossaryNode"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, name: Union["GlossaryNodeUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -856,7 +868,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import AssertionKeyClass
 
 class AssertionUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "assertion"
+    ENTITY_TYPE: ClassVar[Literal["assertion"]] = "assertion"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, assertion_id: Union["AssertionUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -912,7 +924,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import ExecutionRequestKeyClass
 
 class DataHubExecutionRequestUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubExecutionRequest"
+    ENTITY_TYPE: ClassVar[Literal["dataHubExecutionRequest"]] = "dataHubExecutionRequest"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubExecutionRequestUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -968,7 +980,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import MLModelGroupKeyClass
 
 class MlModelGroupUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "mlModelGroup"
+    ENTITY_TYPE: ClassVar[Literal["mlModelGroup"]] = "mlModelGroup"
     _URN_PARTS: ClassVar[int] = 3
 
     def __init__(self, platform: Union["DataPlatformUrn", str], name: str, env: str = "PROD", *, _allow_coercion: bool = True) -> None:
@@ -1031,7 +1043,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import MLModelDeploymentKeyClass
 
 class MlModelDeploymentUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "mlModelDeployment"
+    ENTITY_TYPE: ClassVar[Literal["mlModelDeployment"]] = "mlModelDeployment"
     _URN_PARTS: ClassVar[int] = 3
 
     def __init__(self, platform: Union["DataPlatformUrn", str], name: str, env: str = "PROD", *, _allow_coercion: bool = True) -> None:
@@ -1094,7 +1106,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataFlowKeyClass
 
 class DataFlowUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataFlow"
+    ENTITY_TYPE: ClassVar[Literal["dataFlow"]] = "dataFlow"
     _URN_PARTS: ClassVar[int] = 3
 
     def __init__(self, orchestrator: str, flow_id: str, cluster: str, *, _allow_coercion: bool = True) -> None:
@@ -1183,7 +1195,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataJobKeyClass
 
 class DataJobUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataJob"
+    ENTITY_TYPE: ClassVar[Literal["dataJob"]] = "dataJob"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, flow: Union["DataFlowUrn", str], job_id: str, *, _allow_coercion: bool = True) -> None:
@@ -1255,7 +1267,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import CorpGroupKeyClass
 
 class CorpGroupUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "corpGroup"
+    ENTITY_TYPE: ClassVar[Literal["corpGroup"]] = "corpGroup"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, name: Union["CorpGroupUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1316,7 +1328,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataPlatformKeyClass
 
 class DataPlatformUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataPlatform"
+    ENTITY_TYPE: ClassVar[Literal["dataPlatform"]] = "dataPlatform"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, platform_name: Union["DataPlatformUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1377,7 +1389,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import CorpUserKeyClass
 
 class CorpUserUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "corpuser"
+    ENTITY_TYPE: ClassVar[Literal["corpuser"]] = "corpuser"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, username: Union["CorpUserUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1438,7 +1450,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DashboardKeyClass
 
 class DashboardUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dashboard"
+    ENTITY_TYPE: ClassVar[Literal["dashboard"]] = "dashboard"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, dashboard_tool: str, dashboard_id: str, *, _allow_coercion: bool = True) -> None:
@@ -1480,6 +1492,18 @@ class DashboardUrn(_SpecificUrn):
     def from_key_aspect(cls, key_aspect: "DashboardKeyClass") -> "DashboardUrn":
         return cls(dashboard_tool=key_aspect.dashboardTool, dashboard_id=key_aspect.dashboardId)
 
+    @classmethod
+    def create_from_ids(
+        cls,
+        platform: str,
+        name: str,
+        platform_instance: Optional[str] = None,
+    ) -> "DashboardUrn":
+        return DashboardUrn(
+            dashboard_tool=platform,
+            dashboard_id=f"{platform_instance}.{name}" if platform_instance else name,
+        )
+        
     @property
     def dashboard_tool(self) -> str:
         return self._entity_ids[0]
@@ -1492,7 +1516,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import MLPrimaryKeyKeyClass
 
 class MlPrimaryKeyUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "mlPrimaryKey"
+    ENTITY_TYPE: ClassVar[Literal["mlPrimaryKey"]] = "mlPrimaryKey"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, feature_namespace: str, name: str, *, _allow_coercion: bool = True) -> None:
@@ -1546,7 +1570,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import QueryKeyClass
 
 class QueryUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "query"
+    ENTITY_TYPE: ClassVar[Literal["query"]] = "query"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["QueryUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1602,7 +1626,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import OwnershipTypeKeyClass
 
 class OwnershipTypeUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "ownershipType"
+    ENTITY_TYPE: ClassVar[Literal["ownershipType"]] = "ownershipType"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["OwnershipTypeUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1658,7 +1682,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DomainKeyClass
 
 class DomainUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "domain"
+    ENTITY_TYPE: ClassVar[Literal["domain"]] = "domain"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DomainUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1719,7 +1743,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import MLFeatureKeyClass
 
 class MlFeatureUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "mlFeature"
+    ENTITY_TYPE: ClassVar[Literal["mlFeature"]] = "mlFeature"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, feature_namespace: str, name: str, *, _allow_coercion: bool = True) -> None:
@@ -1773,7 +1797,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import MLFeatureTableKeyClass
 
 class MlFeatureTableUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "mlFeatureTable"
+    ENTITY_TYPE: ClassVar[Literal["mlFeatureTable"]] = "mlFeatureTable"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, platform: Union["DataPlatformUrn", str], name: str, *, _allow_coercion: bool = True) -> None:
@@ -1827,7 +1851,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubViewKeyClass
 
 class DataHubViewUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubView"
+    ENTITY_TYPE: ClassVar[Literal["dataHubView"]] = "dataHubView"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubViewUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1883,7 +1907,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubActionKeyClass
 
 class DataHubActionUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubAction"
+    ENTITY_TYPE: ClassVar[Literal["dataHubAction"]] = "dataHubAction"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubActionUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1939,7 +1963,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubUpgradeKeyClass
 
 class DataHubUpgradeUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubUpgrade"
+    ENTITY_TYPE: ClassVar[Literal["dataHubUpgrade"]] = "dataHubUpgrade"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubUpgradeUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1995,7 +2019,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubStepStateKeyClass
 
 class DataHubStepStateUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubStepState"
+    ENTITY_TYPE: ClassVar[Literal["dataHubStepState"]] = "dataHubStepState"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubStepStateUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2051,7 +2075,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import ERModelRelationshipKeyClass
 
 class ErModelRelationshipUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "erModelRelationship"
+    ENTITY_TYPE: ClassVar[Literal["erModelRelationship"]] = "erModelRelationship"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["ErModelRelationshipUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2107,7 +2131,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubPolicyKeyClass
 
 class DataHubPolicyUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubPolicy"
+    ENTITY_TYPE: ClassVar[Literal["dataHubPolicy"]] = "dataHubPolicy"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubPolicyUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2163,7 +2187,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubRetentionKeyClass
 
 class DataHubRetentionUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubRetention"
+    ENTITY_TYPE: ClassVar[Literal["dataHubRetention"]] = "dataHubRetention"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, entity_name: str, aspect_name: str, *, _allow_coercion: bool = True) -> None:
@@ -2217,7 +2241,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataContractKeyClass
 
 class DataContractUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataContract"
+    ENTITY_TYPE: ClassVar[Literal["dataContract"]] = "dataContract"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataContractUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2273,7 +2297,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataProcessKeyClass
 
 class DataProcessUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataProcess"
+    ENTITY_TYPE: ClassVar[Literal["dataProcess"]] = "dataProcess"
     _URN_PARTS: ClassVar[int] = 3
 
     def __init__(self, name: str, orchestrator: str, env: str = "PROD", *, _allow_coercion: bool = True) -> None:
@@ -2336,7 +2360,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataProcessInstanceKeyClass
 
 class DataProcessInstanceUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataProcessInstance"
+    ENTITY_TYPE: ClassVar[Literal["dataProcessInstance"]] = "dataProcessInstance"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataProcessInstanceUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2401,7 +2425,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import GlossaryTermKeyClass
 
 class GlossaryTermUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "glossaryTerm"
+    ENTITY_TYPE: ClassVar[Literal["glossaryTerm"]] = "glossaryTerm"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, name: Union["GlossaryTermUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2457,7 +2481,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import FormKeyClass
 
 class FormUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "form"
+    ENTITY_TYPE: ClassVar[Literal["form"]] = "form"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["FormUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2513,7 +2537,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import SchemaFieldKeyClass
 
 class SchemaFieldUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "schemaField"
+    ENTITY_TYPE: ClassVar[Literal["schemaField"]] = "schemaField"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, parent: Union["Urn", str], field_path: str, *, _allow_coercion: bool = True) -> None:
@@ -2574,7 +2598,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import TagKeyClass
 
 class TagUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "tag"
+    ENTITY_TYPE: ClassVar[Literal["tag"]] = "tag"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, name: Union["TagUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2635,7 +2659,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import PostKeyClass
 
 class PostUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "post"
+    ENTITY_TYPE: ClassVar[Literal["post"]] = "post"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["PostUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2691,7 +2715,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubIngestionSourceKeyClass
 
 class DataHubIngestionSourceUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubIngestionSource"
+    ENTITY_TYPE: ClassVar[Literal["dataHubIngestionSource"]] = "dataHubIngestionSource"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubIngestionSourceUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2747,7 +2771,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import InviteTokenKeyClass
 
 class InviteTokenUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "inviteToken"
+    ENTITY_TYPE: ClassVar[Literal["inviteToken"]] = "inviteToken"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["InviteTokenUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2803,7 +2827,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubPersonaKeyClass
 
 class DataHubPersonaUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubPersona"
+    ENTITY_TYPE: ClassVar[Literal["dataHubPersona"]] = "dataHubPersona"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubPersonaUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2859,7 +2883,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import ContainerKeyClass
 
 class ContainerUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "container"
+    ENTITY_TYPE: ClassVar[Literal["container"]] = "container"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, guid: Union["ContainerUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2915,7 +2939,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import IncidentKeyClass
 
 class IncidentUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "incident"
+    ENTITY_TYPE: ClassVar[Literal["incident"]] = "incident"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["IncidentUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2971,7 +2995,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataPlatformInstanceKeyClass
 
 class DataPlatformInstanceUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataPlatformInstance"
+    ENTITY_TYPE: ClassVar[Literal["dataPlatformInstance"]] = "dataPlatformInstance"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, platform: Union["DataPlatformUrn", str], instance: str, *, _allow_coercion: bool = True) -> None:
@@ -3025,7 +3049,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import TestKeyClass
 
 class TestUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "test"
+    ENTITY_TYPE: ClassVar[Literal["test"]] = "test"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["TestUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3081,7 +3105,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import VersionSetKeyClass
 
 class VersionSetUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "versionSet"
+    ENTITY_TYPE: ClassVar[Literal["versionSet"]] = "versionSet"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, id: str, entity_type: str, *, _allow_coercion: bool = True) -> None:
@@ -3135,7 +3159,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubSecretKeyClass
 
 class DataHubSecretUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubSecret"
+    ENTITY_TYPE: ClassVar[Literal["dataHubSecret"]] = "dataHubSecret"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubSecretUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3191,7 +3215,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import EntityTypeKeyClass
 
 class EntityTypeUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "entityType"
+    ENTITY_TYPE: ClassVar[Literal["entityType"]] = "entityType"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["EntityTypeUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3247,7 +3271,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import StructuredPropertyKeyClass
 
 class StructuredPropertyUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "structuredProperty"
+    ENTITY_TYPE: ClassVar[Literal["structuredProperty"]] = "structuredProperty"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["StructuredPropertyUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3303,7 +3327,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import PlatformResourceKeyClass
 
 class PlatformResourceUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "platformResource"
+    ENTITY_TYPE: ClassVar[Literal["platformResource"]] = "platformResource"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["PlatformResourceUrn", str], *, _allow_coercion: bool = True) -> None:

datahub/metadata/schema.avsc

@@ -13272,7 +13272,8 @@
         "status",
         "structuredProperties",
         "forms",
-        "testResults"
+        "testResults",
+        "subTypes"
       ]
     },
     "name": "DataProductKey",
@@ -14051,6 +14052,24 @@
         "name": "sortId",
         "doc": "Sort identifier that determines where a version lives in the order of the Version Set.\nWhat this looks like depends on the Version Scheme. For sort ids generated by DataHub we use an 8 character string representation."
       },
+      {
+        "type": {
+          "type": "enum",
+          "symbolDocs": {
+            "ALPHANUMERIC_GENERATED_BY_DATAHUB": "String managed by DataHub. Currently, an 8 character alphabetical string.",
+            "LEXICOGRAPHIC_STRING": "String sorted lexicographically."
+          },
+          "name": "VersioningScheme",
+          "namespace": "com.linkedin.pegasus2avro.versionset",
+          "symbols": [
+            "LEXICOGRAPHIC_STRING",
+            "ALPHANUMERIC_GENERATED_BY_DATAHUB"
+          ]
+        },
+        "name": "versioningScheme",
+        "default": "LEXICOGRAPHIC_STRING",
+        "doc": "What versioning scheme `sortId` belongs to.\nDefaults to a plain string that is lexicographically sorted."
+      },
       {
         "type": [
           "null",
@@ -15834,7 +15853,8 @@
       "keyForEntity": "post",
       "entityCategory": "core",
       "entityAspects": [
-        "postInfo"
+        "postInfo",
+        "subTypes"
       ]
     },
     "name": "PostKey",
@@ -16522,14 +16542,7 @@
         "doc": "The latest versioned entity linked to in this version set"
       },
       {
-        "type": {
-          "type": "enum",
-          "name": "VersioningScheme",
-          "namespace": "com.linkedin.pegasus2avro.versionset",
-          "symbols": [
-            "ALPHANUMERIC_GENERATED_BY_DATAHUB"
-          ]
-        },
+        "type": "com.linkedin.pegasus2avro.versionset.VersioningScheme",
         "name": "versioningScheme",
         "doc": "What versioning scheme is being utilized for the versioned entities sort criterion. Static once set"
       }

datahub/metadata/schemas/CorpGroupKey.avsc

@@ -14,7 +14,8 @@
       "roleMembership",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ],
     "entityDoc": "CorpGroup represents an identity of a group of users in the enterprise."
   },

datahub/metadata/schemas/CorpUserKey.avsc

@@ -18,7 +18,8 @@
       "roleMembership",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ],
     "entityDoc": "CorpUser represents an identity of a person (or an account) in the enterprise."
   },

datahub/metadata/schemas/DataProcessKey.avsc

@@ -8,7 +8,8 @@
       "dataProcessInfo",
       "ownership",
       "status",
-      "testResults"
+      "testResults",
+      "subTypes"
     ]
   },
   "name": "DataProcessKey",

datahub/metadata/schemas/DataProductKey.avsc

@@ -14,7 +14,8 @@
       "status",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ]
   },
   "name": "DataProductKey",

datahub/metadata/schemas/GlossaryNodeKey.avsc

@@ -11,7 +11,8 @@
       "status",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ]
   },
   "name": "GlossaryNodeKey",

datahub/metadata/schemas/GlossaryTermKey.avsc

@@ -16,7 +16,8 @@
       "browsePaths",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ]
   },
   "name": "GlossaryTermKey",