acryl-datahub-cloud 0.3.9rc6__py3-none-any.whl → 0.3.9rc8__py3-none-any.whl

acryl_datahub_cloud/_codegen_config.json

@@ -1,6 +1,6 @@
 {
   "name": "acryl-datahub-cloud",
-  "version": "0.3.9rc6",
+  "version": "0.3.9rc8",
   "install_requires": [
     "avro-gen3==0.7.16",
     "acryl-datahub"

acryl_datahub_cloud/datahub_reporting/datahub_form_reporting.py

@@ -10,7 +10,7 @@ from pydantic import BaseModel
 from acryl_datahub_cloud.elasticsearch.graph_service import BaseModelRow
 from datahub.emitter.mcp import MetadataChangeProposalWrapper
 from datahub.ingestion.graph.client import DataHubGraph
-from datahub.ingestion.graph.filters import SearchFilterRule
+from datahub.ingestion.graph.filters import RawSearchFilterRule
 from datahub.metadata.schema_classes import (
     DomainPropertiesClass,
     FormAssociationClass,
@@ -149,7 +149,7 @@ class DataHubFormReportingData(FormData):
             )
         )
 
-    def get_form_existence_or_filters(self) -> List[SearchFilterRule]:
+    def get_form_existence_or_filters(self) -> List[RawSearchFilterRule]:
         """
         Datasets must either have completedForms or incompleteForms assigned to
         them

acryl_datahub_cloud/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 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:
@@ -81,7 +81,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:
@@ -137,7 +137,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:
@@ -193,7 +193,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:
@@ -249,7 +249,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:
@@ -305,7 +305,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:
@@ -361,7 +361,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:
@@ -417,7 +417,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:
@@ -473,7 +473,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:
@@ -534,7 +534,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:
@@ -595,7 +595,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:
@@ -651,7 +651,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:
@@ -716,7 +716,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:
@@ -772,7 +772,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:
@@ -828,7 +828,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:
@@ -882,7 +882,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:
@@ -938,7 +938,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import RecommendationModuleKeyClass
 
 class RecommendationModuleUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "recommendationModule"
+    ENTITY_TYPE: ClassVar[Literal["recommendationModule"]] = "recommendationModule"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, module_id: str, identifier: str, *, _allow_coercion: bool = True) -> None:
@@ -992,7 +992,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:
@@ -1048,7 +1048,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:
@@ -1102,7 +1102,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:
@@ -1156,7 +1156,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubMetricCubeKeyClass
 
 class DataHubMetricCubeUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubMetricCube"
+    ENTITY_TYPE: ClassVar[Literal["dataHubMetricCube"]] = "dataHubMetricCube"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubMetricCubeUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1212,7 +1212,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:
@@ -1268,7 +1268,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:
@@ -1360,7 +1360,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:
@@ -1416,7 +1416,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:
@@ -1470,7 +1470,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:
@@ -1533,7 +1533,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import SubscriptionKeyClass
 
 class SubscriptionUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "subscription"
+    ENTITY_TYPE: ClassVar[Literal["subscription"]] = "subscription"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["SubscriptionUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1589,7 +1589,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import AnomalyKeyClass
 
 class AnomalyUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "anomaly"
+    ENTITY_TYPE: ClassVar[Literal["anomaly"]] = "anomaly"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["AnomalyUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1645,7 +1645,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import RemoteExecutorPoolKeyClass
 
 class DataHubRemoteExecutorPoolUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubRemoteExecutorPool"
+    ENTITY_TYPE: ClassVar[Literal["dataHubRemoteExecutorPool"]] = "dataHubRemoteExecutorPool"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubRemoteExecutorPoolUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -1701,7 +1701,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:
@@ -1773,7 +1773,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:
@@ -1834,7 +1834,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:
@@ -1890,7 +1890,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:
@@ -1946,7 +1946,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:
@@ -2002,7 +2002,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:
@@ -2058,7 +2058,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:
@@ -2114,7 +2114,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:
@@ -2175,7 +2175,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:
@@ -2231,7 +2231,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:
@@ -2292,7 +2292,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:
@@ -2348,7 +2348,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import ConstraintKeyClass
 
 class ConstraintUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "constraint"
+    ENTITY_TYPE: ClassVar[Literal["constraint"]] = "constraint"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["ConstraintUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -2404,7 +2404,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:
@@ -2460,7 +2460,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:
@@ -2523,7 +2523,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:
@@ -2579,7 +2579,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:
@@ -2641,7 +2641,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:
@@ -2695,7 +2695,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:
@@ -2751,7 +2751,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:
@@ -2805,7 +2805,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:
@@ -2868,7 +2868,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:
@@ -2924,7 +2924,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:
@@ -2980,7 +2980,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:
@@ -3036,7 +3036,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:
@@ -3090,7 +3090,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:
@@ -3146,7 +3146,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import ActionRequestKeyClass
 
 class ActionRequestUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "actionRequest"
+    ENTITY_TYPE: ClassVar[Literal["actionRequest"]] = "actionRequest"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["ActionRequestUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3202,7 +3202,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import RemoteExecutorGlobalConfigKeyClass
 
 class DataHubRemoteExecutorGlobalConfigUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubRemoteExecutorGlobalConfig"
+    ENTITY_TYPE: ClassVar[Literal["dataHubRemoteExecutorGlobalConfig"]] = "dataHubRemoteExecutorGlobalConfig"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubRemoteExecutorGlobalConfigUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3258,7 +3258,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:
@@ -3347,7 +3347,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:
@@ -3403,7 +3403,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:
@@ -3459,7 +3459,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import RemoteExecutorKeyClass
 
 class DataHubRemoteExecutorUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "dataHubRemoteExecutor"
+    ENTITY_TYPE: ClassVar[Literal["dataHubRemoteExecutor"]] = "dataHubRemoteExecutor"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["DataHubRemoteExecutorUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3515,7 +3515,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import MonitorKeyClass
 
 class MonitorUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "monitor"
+    ENTITY_TYPE: ClassVar[Literal["monitor"]] = "monitor"
     _URN_PARTS: ClassVar[int] = 2
 
     def __init__(self, entity: Union["DatasetUrn", str], id: str, *, _allow_coercion: bool = True) -> None:
@@ -3576,7 +3576,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:
@@ -3637,7 +3637,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:
@@ -3691,7 +3691,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:
@@ -3747,7 +3747,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:
@@ -3803,7 +3803,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:
@@ -3866,7 +3866,7 @@ if TYPE_CHECKING:
     from datahub.metadata.schema_classes import LinkPreviewKeyClass
 
 class LinkPreviewUrn(_SpecificUrn):
-    ENTITY_TYPE: ClassVar[str] = "linkPreview"
+    ENTITY_TYPE: ClassVar[Literal["linkPreview"]] = "linkPreview"
     _URN_PARTS: ClassVar[int] = 1
 
     def __init__(self, id: Union["LinkPreviewUrn", str], *, _allow_coercion: bool = True) -> None:
@@ -3922,7 +3922,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: