acryl-datahub 1.0.0rc8__py3-none-any.whl → 1.0.0rc9__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:
@@ -800,7 +800,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 +856,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 +912,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 +968,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 +1031,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 +1094,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 +1183,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 +1255,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 +1316,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 +1377,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 +1438,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:
@@ -1492,7 +1492,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 +1546,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 +1602,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 +1658,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 +1719,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 +1773,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 +1827,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 +1883,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 +1939,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 +1995,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 +2051,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 +2107,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 +2163,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 +2217,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 +2273,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 +2336,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 +2401,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 +2457,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 +2513,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 +2574,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 +2635,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 +2691,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 +2747,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 +2803,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 +2859,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 +2915,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 +2971,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 +3025,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 +3081,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 +3135,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 +3191,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 +3247,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 +3303,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",

datahub/metadata/schemas/MLFeatureKey.avsc

@@ -19,7 +19,8 @@
       "browsePathsV2",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ]
   },
   "name": "MLFeatureKey",

datahub/metadata/schemas/MLFeatureTableKey.avsc

@@ -19,7 +19,8 @@
       "browsePathsV2",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ]
   },
   "name": "MLFeatureTableKey",

datahub/metadata/schemas/MLModelGroupKey.avsc

@@ -18,7 +18,8 @@
       "browsePathsV2",
       "structuredProperties",
       "forms",
-      "testResults"
+      "testResults",
+      "subTypes"
     ]
   },
   "name": "MLModelGroupKey",