acryl-datahub 1.0.0rc5__py3-none-any.whl → 1.0.0rc6__py3-none-any.whl

datahub/metadata/_schema_classes.py

@@ -13935,6 +13935,46 @@ class RoleMembershipClass(_Aspect):
         self._inner_dict['roles'] = value
     
     
+class IncidentAssigneeClass(DictWrapper):
+    """The incident assignee type.
+    This is in a record so that we can add additional fields if we need to later (e.g.
+    the type of the assignee."""
+    
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.incident.IncidentAssignee")
+    def __init__(self,
+        actor: str,
+        assignedAt: "AuditStampClass",
+    ):
+        super().__init__()
+        
+        self.actor = actor
+        self.assignedAt = assignedAt
+    
+    def _restore_defaults(self) -> None:
+        self.actor = str()
+        self.assignedAt = AuditStampClass._construct_with_defaults()
+    
+    
+    @property
+    def actor(self) -> str:
+        """The user or group assigned to the incident."""
+        return self._inner_dict.get('actor')  # type: ignore
+    
+    @actor.setter
+    def actor(self, value: str) -> None:
+        self._inner_dict['actor'] = value
+    
+    
+    @property
+    def assignedAt(self) -> "AuditStampClass":
+        """The time & actor responsible for assiging the assignee."""
+        return self._inner_dict.get('assignedAt')  # type: ignore
+    
+    @assignedAt.setter
+    def assignedAt(self, value: "AuditStampClass") -> None:
+        self._inner_dict['assignedAt'] = value
+    
+    
 class IncidentInfoClass(_Aspect):
     """Information about an incident raised on an asset."""
 
@@ -13952,7 +13992,9 @@ class IncidentInfoClass(_Aspect):
         title: Union[None, str]=None,
         description: Union[None, str]=None,
         priority: Optional[Union[int, None]]=None,
+        assignees: Union[None, List["IncidentAssigneeClass"]]=None,
         source: Union[None, "IncidentSourceClass"]=None,
+        startedAt: Union[None, int]=None,
     ):
         super().__init__()
         
@@ -13966,8 +14008,10 @@ class IncidentInfoClass(_Aspect):
             self.priority = self.RECORD_SCHEMA.fields_dict["priority"].default
         else:
             self.priority = priority
+        self.assignees = assignees
         self.status = status
         self.source = source
+        self.startedAt = startedAt
         self.created = created
     
     def _restore_defaults(self) -> None:
@@ -13977,8 +14021,10 @@ class IncidentInfoClass(_Aspect):
         self.description = self.RECORD_SCHEMA.fields_dict["description"].default
         self.entities = list()
         self.priority = self.RECORD_SCHEMA.fields_dict["priority"].default
+        self.assignees = self.RECORD_SCHEMA.fields_dict["assignees"].default
         self.status = IncidentStatusClass._construct_with_defaults()
         self.source = self.RECORD_SCHEMA.fields_dict["source"].default
+        self.startedAt = self.RECORD_SCHEMA.fields_dict["startedAt"].default
         self.created = AuditStampClass._construct_with_defaults()
     
     
@@ -14034,7 +14080,9 @@ class IncidentInfoClass(_Aspect):
     
     @property
     def priority(self) -> Union[int, None]:
-        """A numeric severity or priority for the incident. On the UI we will translate this into something easy to understand."""
+        """A numeric severity or priority for the incident. On the UI we will translate this into something easy to understand.
+    Currently supported: 0 - CRITICAL, 1 - HIGH, 2 - MED, 3 - LOW
+    (We probably should have modeled as an enum)"""
         return self._inner_dict.get('priority')  # type: ignore
     
     @priority.setter
@@ -14042,6 +14090,16 @@ class IncidentInfoClass(_Aspect):
         self._inner_dict['priority'] = value
     
     
+    @property
+    def assignees(self) -> Union[None, List["IncidentAssigneeClass"]]:
+        """The parties assigned with resolving the incident"""
+        return self._inner_dict.get('assignees')  # type: ignore
+    
+    @assignees.setter
+    def assignees(self, value: Union[None, List["IncidentAssigneeClass"]]) -> None:
+        self._inner_dict['assignees'] = value
+    
+    
     @property
     def status(self) -> "IncidentStatusClass":
         """The current status of an incident, i.e. active or inactive."""
@@ -14062,6 +14120,16 @@ class IncidentInfoClass(_Aspect):
         self._inner_dict['source'] = value
     
     
+    @property
+    def startedAt(self) -> Union[None, int]:
+        """The time at which the incident actually started (may be before the date it was raised)."""
+        return self._inner_dict.get('startedAt')  # type: ignore
+    
+    @startedAt.setter
+    def startedAt(self, value: Union[None, int]) -> None:
+        self._inner_dict['startedAt'] = value
+    
+    
     @property
     def created(self) -> "AuditStampClass":
         """The time at which the request was initially created"""
@@ -14125,6 +14193,27 @@ class IncidentSourceTypeClass(object):
     
     
     
+class IncidentStageClass(object):
+    # No docs available.
+    
+    TRIAGE = "TRIAGE"
+    """The impact and priority of the incident is being actively assessed."""
+    
+    INVESTIGATION = "INVESTIGATION"
+    """The incident root cause is being investigated."""
+    
+    WORK_IN_PROGRESS = "WORK_IN_PROGRESS"
+    """The incident is in the remediation stage."""
+    
+    FIXED = "FIXED"
+    """The incident is in the resolved as completed stage."""
+    
+    NO_ACTION_REQUIRED = "NO_ACTION_REQUIRED"
+    """The incident is in the resolved with no action required state, e.g. the
+    incident was a false positive, or was expected."""
+    
+    
+    
 class IncidentStateClass(object):
     # No docs available.
     
@@ -14143,23 +14232,26 @@ class IncidentStatusClass(DictWrapper):
     def __init__(self,
         state: Union[str, "IncidentStateClass"],
         lastUpdated: "AuditStampClass",
+        stage: Union[None, Union[str, "IncidentStageClass"]]=None,
         message: Union[None, str]=None,
     ):
         super().__init__()
         
         self.state = state
+        self.stage = stage
         self.message = message
         self.lastUpdated = lastUpdated
     
     def _restore_defaults(self) -> None:
         self.state = IncidentStateClass.ACTIVE
+        self.stage = self.RECORD_SCHEMA.fields_dict["stage"].default
         self.message = self.RECORD_SCHEMA.fields_dict["message"].default
         self.lastUpdated = AuditStampClass._construct_with_defaults()
     
     
     @property
     def state(self) -> Union[str, "IncidentStateClass"]:
-        """The state of the incident"""
+        """The top-level state of the incident, whether it's active or resolved."""
         return self._inner_dict.get('state')  # type: ignore
     
     @state.setter
@@ -14167,6 +14259,17 @@ class IncidentStatusClass(DictWrapper):
         self._inner_dict['state'] = value
     
     
+    @property
+    def stage(self) -> Union[None, Union[str, "IncidentStageClass"]]:
+        """The lifecycle stage for the incident - Null means no stage was assigned yet.
+    In the future, we may add CUSTOM here with a customStage string field for user-defined stages."""
+        return self._inner_dict.get('stage')  # type: ignore
+    
+    @stage.setter
+    def stage(self, value: Union[None, Union[str, "IncidentStageClass"]]) -> None:
+        self._inner_dict['stage'] = value
+    
+    
     @property
     def message(self) -> Union[None, str]:
         """Optional message associated with the incident"""
@@ -25490,9 +25593,11 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.identity.InviteToken': InviteTokenClass,
     'com.linkedin.pegasus2avro.identity.NativeGroupMembership': NativeGroupMembershipClass,
     'com.linkedin.pegasus2avro.identity.RoleMembership': RoleMembershipClass,
+    'com.linkedin.pegasus2avro.incident.IncidentAssignee': IncidentAssigneeClass,
     'com.linkedin.pegasus2avro.incident.IncidentInfo': IncidentInfoClass,
     'com.linkedin.pegasus2avro.incident.IncidentSource': IncidentSourceClass,
     'com.linkedin.pegasus2avro.incident.IncidentSourceType': IncidentSourceTypeClass,
+    'com.linkedin.pegasus2avro.incident.IncidentStage': IncidentStageClass,
     'com.linkedin.pegasus2avro.incident.IncidentState': IncidentStateClass,
     'com.linkedin.pegasus2avro.incident.IncidentStatus': IncidentStatusClass,
     'com.linkedin.pegasus2avro.incident.IncidentType': IncidentTypeClass,
@@ -25968,9 +26073,11 @@ __SCHEMA_TYPES = {
     'InviteToken': InviteTokenClass,
     'NativeGroupMembership': NativeGroupMembershipClass,
     'RoleMembership': RoleMembershipClass,
+    'IncidentAssignee': IncidentAssigneeClass,
     'IncidentInfo': IncidentInfoClass,
     'IncidentSource': IncidentSourceClass,
     'IncidentSourceType': IncidentSourceTypeClass,
+    'IncidentStage': IncidentStageClass,
     'IncidentState': IncidentStateClass,
     'IncidentStatus': IncidentStatusClass,
     'IncidentType': IncidentTypeClass,
@@ -26215,217 +26322,217 @@ avrojson.set_global_json_converter(_json_converter)
     
 
 ASPECT_CLASSES: List[Type[_Aspect]] = [
+    IcebergCatalogInfoClass,
+    DatasetDeprecationClass,
+    ViewPropertiesClass,
+    PartitionsSummaryClass,
+    DatasetUpstreamLineageClass,
+    DatasetPropertiesClass,
+    UpstreamLineageClass,
+    EditableDatasetPropertiesClass,
+    DatasetProfileClass,
+    DatasetUsageStatisticsClass,
+    GlobalSettingsInfoClass,
+    DataHubStepStatePropertiesClass,
+    BusinessAttributeInfoClass,
+    BusinessAttributesClass,
+    BusinessAttributeKeyClass,
+    QuerySubjectsClass,
+    QueryPropertiesClass,
+    QueryUsageStatisticsClass,
+    DataHubViewInfoClass,
+    DataTypeInfoClass,
+    DataTypeKeyClass,
+    FormInfoClass,
+    DynamicFormAssignmentClass,
+    DataHubConnectionDetailsClass,
+    ExecutionRequestSignalClass,
+    ExecutionRequestInputClass,
+    ExecutionRequestResultClass,
+    GlossaryNodeInfoClass,
+    GlossaryTermInfoClass,
+    GlossaryRelatedTermsClass,
+    TelemetryClientIdClass,
     DataHubAccessTokenInfoClass,
-    DataHubRoleInfoClass,
-    DataHubPolicyInfoClass,
-    IncidentSourceClass,
-    IncidentInfoClass,
+    ContainerClass,
+    ContainerPropertiesClass,
+    EditableContainerPropertiesClass,
+    DataHubSecretValueClass,
     DataHubUpgradeRequestClass,
     DataHubUpgradeResultClass,
-    CorpUserCredentialsClass,
-    CorpUserStatusClass,
-    CorpUserInfoClass,
-    InviteTokenClass,
-    CorpUserEditableInfoClass,
-    RoleMembershipClass,
-    CorpGroupEditableInfoClass,
-    CorpGroupInfoClass,
-    GroupMembershipClass,
-    NativeGroupMembershipClass,
-    CorpUserSettingsClass,
-    DataHubRetentionConfigClass,
-    EditableNotebookPropertiesClass,
-    NotebookContentClass,
-    NotebookInfoClass,
-    ExecutionRequestResultClass,
-    ExecutionRequestInputClass,
-    ExecutionRequestSignalClass,
-    DataHubStepStatePropertiesClass,
-    DataTransformLogicClass,
-    OwnershipClass,
+    AssertionRunEventClass,
+    AssertionActionsClass,
+    AssertionInfoClass,
+    EditableSchemaMetadataClass,
+    SchemaMetadataClass,
+    DataProductPropertiesClass,
+    DataProductKeyClass,
+    DataContractStatusClass,
+    DataContractPropertiesClass,
+    OperationClass,
+    StatusClass,
+    InstitutionalMemoryClass,
+    AccessClass,
+    DataPlatformInstanceClass,
     SiblingsClass,
-    BrowsePathsV2Class,
     IncidentsSummaryClass,
-    FormsClass,
-    CostClass,
-    OriginClass,
-    StatusClass,
     EmbedClass,
-    DataPlatformInstanceClass,
-    DocumentationClass,
+    OwnershipClass,
+    SubTypesClass,
+    CostClass,
+    BrowsePathsClass,
     VersionPropertiesClass,
-    GlossaryTermsClass,
+    OriginClass,
     DeprecationClass,
-    AccessClass,
-    InstitutionalMemoryClass,
-    InputFieldsClass,
+    GlossaryTermsClass,
     DisplayPropertiesClass,
-    OperationClass,
+    BrowsePathsV2Class,
     GlobalTagsClass,
-    BrowsePathsClass,
-    SubTypesClass,
-    AssertionInfoClass,
-    AssertionRunEventClass,
-    AssertionActionsClass,
+    DataTransformLogicClass,
+    DocumentationClass,
+    InputFieldsClass,
+    FormsClass,
     DataPlatformInfoClass,
-    IcebergWarehouseInfoClass,
-    DataPlatformInstancePropertiesClass,
-    OwnershipTypeInfoClass,
-    ERModelRelationshipKeyClass,
-    DataProcessInstanceKeyClass,
-    ChartKeyClass,
-    VersionSetKeyClass,
-    DataHubUpgradeKeyClass,
+    DataHubPolicyInfoClass,
+    DataHubRoleInfoClass,
+    IncidentSourceClass,
+    IncidentInfoClass,
+    VersionInfoClass,
+    DataJobInfoClass,
+    EditableDataFlowPropertiesClass,
+    EditableDataJobPropertiesClass,
+    DataFlowInfoClass,
+    DataJobInputOutputClass,
+    DatahubIngestionCheckpointClass,
+    DatahubIngestionRunSummaryClass,
+    TelemetryKeyClass,
     DataHubAccessTokenKeyClass,
-    MLPrimaryKeyKeyClass,
-    OwnershipTypeKeyClass,
-    SchemaFieldKeyClass,
-    PostKeyClass,
-    DataHubSecretKeyClass,
-    DashboardKeyClass,
-    MLModelGroupKeyClass,
-    DatasetKeyClass,
+    DataHubConnectionKeyClass,
+    DataHubRoleKeyClass,
+    MLModelKeyClass,
     NotebookKeyClass,
-    DataFlowKeyClass,
-    MLModelDeploymentKeyClass,
-    FormKeyClass,
-    DomainKeyClass,
-    IncidentKeyClass,
-    TagKeyClass,
     RoleKeyClass,
-    DataPlatformKeyClass,
-    DataJobKeyClass,
-    DataHubStepStateKeyClass,
-    DataHubViewKeyClass,
-    CorpUserKeyClass,
-    DataHubRetentionKeyClass,
-    DataPlatformInstanceKeyClass,
-    DataHubPersonaKeyClass,
-    ContainerKeyClass,
-    GlossaryNodeKeyClass,
-    DataHubIngestionSourceKeyClass,
-    TestKeyClass,
-    DataHubPolicyKeyClass,
-    CorpGroupKeyClass,
-    DataContractKeyClass,
     GlobalSettingsKeyClass,
+    DatasetKeyClass,
+    ChartKeyClass,
+    GlossaryNodeKeyClass,
     AssertionKeyClass,
+    ExecutionRequestKeyClass,
+    MLModelGroupKeyClass,
+    MLModelDeploymentKeyClass,
+    DataFlowKeyClass,
+    DataJobKeyClass,
+    CorpGroupKeyClass,
+    DataPlatformKeyClass,
+    CorpUserKeyClass,
+    DashboardKeyClass,
+    MLPrimaryKeyKeyClass,
     QueryKeyClass,
+    OwnershipTypeKeyClass,
+    DomainKeyClass,
+    MLFeatureKeyClass,
+    MLFeatureTableKeyClass,
+    DataHubViewKeyClass,
     DataHubActionKeyClass,
-    MLModelKeyClass,
-    ExecutionRequestKeyClass,
-    InviteTokenKeyClass,
+    DataHubUpgradeKeyClass,
+    DataHubStepStateKeyClass,
+    ERModelRelationshipKeyClass,
+    DataHubPolicyKeyClass,
+    DataHubRetentionKeyClass,
+    DataContractKeyClass,
     DataProcessKeyClass,
+    DataProcessInstanceKeyClass,
     GlossaryTermKeyClass,
-    MLFeatureTableKeyClass,
-    MLFeatureKeyClass,
-    DataHubRoleKeyClass,
-    TelemetryKeyClass,
-    DataHubConnectionKeyClass,
-    DataJobInfoClass,
-    DataJobInputOutputClass,
-    VersionInfoClass,
-    EditableDataFlowPropertiesClass,
-    DataFlowInfoClass,
-    EditableDataJobPropertiesClass,
-    DatahubIngestionCheckpointClass,
-    DatahubIngestionRunSummaryClass,
-    BusinessAttributeInfoClass,
-    BusinessAttributeKeyClass,
-    BusinessAttributesClass,
-    DataHubSecretValueClass,
-    GlossaryTermInfoClass,
-    GlossaryNodeInfoClass,
-    GlossaryRelatedTermsClass,
-    MLModelGroupPropertiesClass,
-    IntendedUseClass,
-    EthicalConsiderationsClass,
-    CaveatsAndRecommendationsClass,
-    MLMetricClass,
-    MLHyperParamClass,
-    MLTrainingRunPropertiesClass,
-    EditableMLFeaturePropertiesClass,
-    TrainingDataClass,
-    QuantitativeAnalysesClass,
-    MLModelDeploymentPropertiesClass,
-    MetricsClass,
-    EditableMLPrimaryKeyPropertiesClass,
-    MLFeaturePropertiesClass,
-    EditableMLModelPropertiesClass,
-    MLModelPropertiesClass,
-    EditableMLFeatureTablePropertiesClass,
-    SourceCodeClass,
-    EditableMLModelGroupPropertiesClass,
-    MLPrimaryKeyPropertiesClass,
-    MLModelFactorPromptsClass,
-    EvaluationDataClass,
-    MLFeatureTablePropertiesClass,
-    DataHubViewInfoClass,
-    TelemetryClientIdClass,
-    DashboardUsageStatisticsClass,
-    EditableDashboardPropertiesClass,
-    DashboardInfoClass,
-    DataProcessInstanceInputClass,
+    FormKeyClass,
+    SchemaFieldKeyClass,
+    TagKeyClass,
+    PostKeyClass,
+    DataHubIngestionSourceKeyClass,
+    InviteTokenKeyClass,
+    DataHubPersonaKeyClass,
+    ContainerKeyClass,
+    IncidentKeyClass,
+    DataPlatformInstanceKeyClass,
+    TestKeyClass,
+    VersionSetKeyClass,
+    DataHubSecretKeyClass,
+    DataHubPersonaInfoClass,
+    DataPlatformInstancePropertiesClass,
+    IcebergWarehouseInfoClass,
+    DomainsClass,
+    DomainPropertiesClass,
+    OwnershipTypeInfoClass,
+    SchemaFieldInfoClass,
+    SchemaFieldAliasesClass,
+    VersionSetPropertiesClass,
+    TestResultsClass,
+    TestInfoClass,
+    DataProcessInfoClass,
     DataProcessInstanceOutputClass,
     DataProcessInstanceRelationshipsClass,
-    DataProcessInfoClass,
-    DataProcessInstanceRunEventClass,
     DataProcessInstancePropertiesClass,
-    DataTypeKeyClass,
-    DataTypeInfoClass,
-    FormInfoClass,
-    DynamicFormAssignmentClass,
-    DataHubConnectionDetailsClass,
-    DataProductKeyClass,
-    DataProductPropertiesClass,
-    DataHubPersonaInfoClass,
-    EntityTypeInfoClass,
+    DataProcessInstanceInputClass,
+    DataProcessInstanceRunEventClass,
     EntityTypeKeyClass,
-    PostInfoClass,
+    EntityTypeInfoClass,
+    EditableERModelRelationshipPropertiesClass,
+    ERModelRelationshipPropertiesClass,
+    CorpUserStatusClass,
+    CorpUserCredentialsClass,
+    CorpUserEditableInfoClass,
+    CorpGroupInfoClass,
+    CorpGroupEditableInfoClass,
+    CorpUserInfoClass,
+    CorpUserSettingsClass,
+    RoleMembershipClass,
+    GroupMembershipClass,
+    NativeGroupMembershipClass,
+    InviteTokenClass,
     RolePropertiesClass,
     ActorsClass,
-    DataContractPropertiesClass,
-    DataContractStatusClass,
-    ERModelRelationshipPropertiesClass,
-    EditableERModelRelationshipPropertiesClass,
-    ChartUsageStatisticsClass,
-    ChartQueryClass,
-    ChartInfoClass,
-    EditableChartPropertiesClass,
-    QueryPropertiesClass,
-    QuerySubjectsClass,
-    QueryUsageStatisticsClass,
-    VersionSetPropertiesClass,
+    PostInfoClass,
+    EditableDashboardPropertiesClass,
+    DashboardInfoClass,
+    DashboardUsageStatisticsClass,
+    NotebookInfoClass,
+    NotebookContentClass,
+    EditableNotebookPropertiesClass,
     DataHubIngestionSourceInfoClass,
-    TagPropertiesClass,
-    DomainsClass,
-    DomainPropertiesClass,
-    EditableSchemaMetadataClass,
-    SchemaMetadataClass,
-    GlobalSettingsInfoClass,
-    ViewPropertiesClass,
-    DatasetUpstreamLineageClass,
-    DatasetProfileClass,
-    EditableDatasetPropertiesClass,
-    IcebergCatalogInfoClass,
-    DatasetUsageStatisticsClass,
-    PartitionsSummaryClass,
-    DatasetPropertiesClass,
-    UpstreamLineageClass,
-    DatasetDeprecationClass,
-    EditableContainerPropertiesClass,
-    ContainerClass,
-    ContainerPropertiesClass,
-    SchemaFieldAliasesClass,
-    SchemaFieldInfoClass,
-    PlatformResourceKeyClass,
-    PlatformResourceInfoClass,
-    TestInfoClass,
-    TestResultsClass,
     StructuredPropertiesClass,
     StructuredPropertyDefinitionClass,
     StructuredPropertyKeyClass,
-    StructuredPropertySettingsClass
+    StructuredPropertySettingsClass,
+    TagPropertiesClass,
+    EthicalConsiderationsClass,
+    MLModelDeploymentPropertiesClass,
+    MLPrimaryKeyPropertiesClass,
+    TrainingDataClass,
+    MLModelFactorPromptsClass,
+    QuantitativeAnalysesClass,
+    MLHyperParamClass,
+    EditableMLFeatureTablePropertiesClass,
+    EditableMLPrimaryKeyPropertiesClass,
+    SourceCodeClass,
+    EvaluationDataClass,
+    MLTrainingRunPropertiesClass,
+    EditableMLModelGroupPropertiesClass,
+    MLFeatureTablePropertiesClass,
+    MetricsClass,
+    MLMetricClass,
+    CaveatsAndRecommendationsClass,
+    MLModelGroupPropertiesClass,
+    MLFeaturePropertiesClass,
+    IntendedUseClass,
+    EditableMLFeaturePropertiesClass,
+    MLModelPropertiesClass,
+    EditableMLModelPropertiesClass,
+    PlatformResourceKeyClass,
+    PlatformResourceInfoClass,
+    ChartUsageStatisticsClass,
+    ChartQueryClass,
+    EditableChartPropertiesClass,
+    ChartInfoClass,
+    DataHubRetentionConfigClass
 ]
 
 ASPECT_NAME_MAP: Dict[str, Type[_Aspect]] = {
@@ -26436,277 +26543,277 @@ ASPECT_NAME_MAP: Dict[str, Type[_Aspect]] = {
 from typing_extensions import TypedDict
 
 class AspectBag(TypedDict, total=False):
+    icebergCatalogInfo: IcebergCatalogInfoClass
+    datasetDeprecation: DatasetDeprecationClass
+    viewProperties: ViewPropertiesClass
+    partitionsSummary: PartitionsSummaryClass
+    datasetUpstreamLineage: DatasetUpstreamLineageClass
+    datasetProperties: DatasetPropertiesClass
+    upstreamLineage: UpstreamLineageClass
+    editableDatasetProperties: EditableDatasetPropertiesClass
+    datasetProfile: DatasetProfileClass
+    datasetUsageStatistics: DatasetUsageStatisticsClass
+    globalSettingsInfo: GlobalSettingsInfoClass
+    dataHubStepStateProperties: DataHubStepStatePropertiesClass
+    businessAttributeInfo: BusinessAttributeInfoClass
+    businessAttributes: BusinessAttributesClass
+    businessAttributeKey: BusinessAttributeKeyClass
+    querySubjects: QuerySubjectsClass
+    queryProperties: QueryPropertiesClass
+    queryUsageStatistics: QueryUsageStatisticsClass
+    dataHubViewInfo: DataHubViewInfoClass
+    dataTypeInfo: DataTypeInfoClass
+    dataTypeKey: DataTypeKeyClass
+    formInfo: FormInfoClass
+    dynamicFormAssignment: DynamicFormAssignmentClass
+    dataHubConnectionDetails: DataHubConnectionDetailsClass
+    dataHubExecutionRequestSignal: ExecutionRequestSignalClass
+    dataHubExecutionRequestInput: ExecutionRequestInputClass
+    dataHubExecutionRequestResult: ExecutionRequestResultClass
+    glossaryNodeInfo: GlossaryNodeInfoClass
+    glossaryTermInfo: GlossaryTermInfoClass
+    glossaryRelatedTerms: GlossaryRelatedTermsClass
+    telemetryClientId: TelemetryClientIdClass
     dataHubAccessTokenInfo: DataHubAccessTokenInfoClass
-    dataHubRoleInfo: DataHubRoleInfoClass
-    dataHubPolicyInfo: DataHubPolicyInfoClass
-    incidentSource: IncidentSourceClass
-    incidentInfo: IncidentInfoClass
+    container: ContainerClass
+    containerProperties: ContainerPropertiesClass
+    editableContainerProperties: EditableContainerPropertiesClass
+    dataHubSecretValue: DataHubSecretValueClass
     dataHubUpgradeRequest: DataHubUpgradeRequestClass
     dataHubUpgradeResult: DataHubUpgradeResultClass
-    corpUserCredentials: CorpUserCredentialsClass
-    corpUserStatus: CorpUserStatusClass
-    corpUserInfo: CorpUserInfoClass
-    inviteToken: InviteTokenClass
-    corpUserEditableInfo: CorpUserEditableInfoClass
-    roleMembership: RoleMembershipClass
-    corpGroupEditableInfo: CorpGroupEditableInfoClass
-    corpGroupInfo: CorpGroupInfoClass
-    groupMembership: GroupMembershipClass
-    nativeGroupMembership: NativeGroupMembershipClass
-    corpUserSettings: CorpUserSettingsClass
-    dataHubRetentionConfig: DataHubRetentionConfigClass
-    editableNotebookProperties: EditableNotebookPropertiesClass
-    notebookContent: NotebookContentClass
-    notebookInfo: NotebookInfoClass
-    dataHubExecutionRequestResult: ExecutionRequestResultClass
-    dataHubExecutionRequestInput: ExecutionRequestInputClass
-    dataHubExecutionRequestSignal: ExecutionRequestSignalClass
-    dataHubStepStateProperties: DataHubStepStatePropertiesClass
-    dataTransformLogic: DataTransformLogicClass
-    ownership: OwnershipClass
+    assertionRunEvent: AssertionRunEventClass
+    assertionActions: AssertionActionsClass
+    assertionInfo: AssertionInfoClass
+    editableSchemaMetadata: EditableSchemaMetadataClass
+    schemaMetadata: SchemaMetadataClass
+    dataProductProperties: DataProductPropertiesClass
+    dataProductKey: DataProductKeyClass
+    dataContractStatus: DataContractStatusClass
+    dataContractProperties: DataContractPropertiesClass
+    operation: OperationClass
+    status: StatusClass
+    institutionalMemory: InstitutionalMemoryClass
+    access: AccessClass
+    dataPlatformInstance: DataPlatformInstanceClass
     siblings: SiblingsClass
-    browsePathsV2: BrowsePathsV2Class
     incidentsSummary: IncidentsSummaryClass
-    forms: FormsClass
-    cost: CostClass
-    origin: OriginClass
-    status: StatusClass
     embed: EmbedClass
-    dataPlatformInstance: DataPlatformInstanceClass
-    documentation: DocumentationClass
+    ownership: OwnershipClass
+    subTypes: SubTypesClass
+    cost: CostClass
+    browsePaths: BrowsePathsClass
     versionProperties: VersionPropertiesClass
-    glossaryTerms: GlossaryTermsClass
+    origin: OriginClass
     deprecation: DeprecationClass
-    access: AccessClass
-    institutionalMemory: InstitutionalMemoryClass
-    inputFields: InputFieldsClass
+    glossaryTerms: GlossaryTermsClass
     displayProperties: DisplayPropertiesClass
-    operation: OperationClass
+    browsePathsV2: BrowsePathsV2Class
     globalTags: GlobalTagsClass
-    browsePaths: BrowsePathsClass
-    subTypes: SubTypesClass
-    assertionInfo: AssertionInfoClass
-    assertionRunEvent: AssertionRunEventClass
-    assertionActions: AssertionActionsClass
+    dataTransformLogic: DataTransformLogicClass
+    documentation: DocumentationClass
+    inputFields: InputFieldsClass
+    forms: FormsClass
     dataPlatformInfo: DataPlatformInfoClass
-    icebergWarehouseInfo: IcebergWarehouseInfoClass
-    dataPlatformInstanceProperties: DataPlatformInstancePropertiesClass
-    ownershipTypeInfo: OwnershipTypeInfoClass
-    erModelRelationshipKey: ERModelRelationshipKeyClass
-    dataProcessInstanceKey: DataProcessInstanceKeyClass
-    chartKey: ChartKeyClass
-    versionSetKey: VersionSetKeyClass
-    dataHubUpgradeKey: DataHubUpgradeKeyClass
+    dataHubPolicyInfo: DataHubPolicyInfoClass
+    dataHubRoleInfo: DataHubRoleInfoClass
+    incidentSource: IncidentSourceClass
+    incidentInfo: IncidentInfoClass
+    versionInfo: VersionInfoClass
+    dataJobInfo: DataJobInfoClass
+    editableDataFlowProperties: EditableDataFlowPropertiesClass
+    editableDataJobProperties: EditableDataJobPropertiesClass
+    dataFlowInfo: DataFlowInfoClass
+    dataJobInputOutput: DataJobInputOutputClass
+    datahubIngestionCheckpoint: DatahubIngestionCheckpointClass
+    datahubIngestionRunSummary: DatahubIngestionRunSummaryClass
+    telemetryKey: TelemetryKeyClass
     dataHubAccessTokenKey: DataHubAccessTokenKeyClass
-    mlPrimaryKeyKey: MLPrimaryKeyKeyClass
-    ownershipTypeKey: OwnershipTypeKeyClass
-    schemaFieldKey: SchemaFieldKeyClass
-    postKey: PostKeyClass
-    dataHubSecretKey: DataHubSecretKeyClass
-    dashboardKey: DashboardKeyClass
-    mlModelGroupKey: MLModelGroupKeyClass
-    datasetKey: DatasetKeyClass
+    dataHubConnectionKey: DataHubConnectionKeyClass
+    dataHubRoleKey: DataHubRoleKeyClass
+    mlModelKey: MLModelKeyClass
     notebookKey: NotebookKeyClass
-    dataFlowKey: DataFlowKeyClass
-    mlModelDeploymentKey: MLModelDeploymentKeyClass
-    formKey: FormKeyClass
-    domainKey: DomainKeyClass
-    incidentKey: IncidentKeyClass
-    tagKey: TagKeyClass
     roleKey: RoleKeyClass
-    dataPlatformKey: DataPlatformKeyClass
-    dataJobKey: DataJobKeyClass
-    dataHubStepStateKey: DataHubStepStateKeyClass
-    dataHubViewKey: DataHubViewKeyClass
-    corpUserKey: CorpUserKeyClass
-    dataHubRetentionKey: DataHubRetentionKeyClass
-    dataPlatformInstanceKey: DataPlatformInstanceKeyClass
-    dataHubPersonaKey: DataHubPersonaKeyClass
-    containerKey: ContainerKeyClass
-    glossaryNodeKey: GlossaryNodeKeyClass
-    dataHubIngestionSourceKey: DataHubIngestionSourceKeyClass
-    testKey: TestKeyClass
-    dataHubPolicyKey: DataHubPolicyKeyClass
-    corpGroupKey: CorpGroupKeyClass
-    dataContractKey: DataContractKeyClass
     globalSettingsKey: GlobalSettingsKeyClass
+    datasetKey: DatasetKeyClass
+    chartKey: ChartKeyClass
+    glossaryNodeKey: GlossaryNodeKeyClass
     assertionKey: AssertionKeyClass
+    dataHubExecutionRequestKey: ExecutionRequestKeyClass
+    mlModelGroupKey: MLModelGroupKeyClass
+    mlModelDeploymentKey: MLModelDeploymentKeyClass
+    dataFlowKey: DataFlowKeyClass
+    dataJobKey: DataJobKeyClass
+    corpGroupKey: CorpGroupKeyClass
+    dataPlatformKey: DataPlatformKeyClass
+    corpUserKey: CorpUserKeyClass
+    dashboardKey: DashboardKeyClass
+    mlPrimaryKeyKey: MLPrimaryKeyKeyClass
     queryKey: QueryKeyClass
+    ownershipTypeKey: OwnershipTypeKeyClass
+    domainKey: DomainKeyClass
+    mlFeatureKey: MLFeatureKeyClass
+    mlFeatureTableKey: MLFeatureTableKeyClass
+    dataHubViewKey: DataHubViewKeyClass
     dataHubActionKey: DataHubActionKeyClass
-    mlModelKey: MLModelKeyClass
-    dataHubExecutionRequestKey: ExecutionRequestKeyClass
-    inviteTokenKey: InviteTokenKeyClass
+    dataHubUpgradeKey: DataHubUpgradeKeyClass
+    dataHubStepStateKey: DataHubStepStateKeyClass
+    erModelRelationshipKey: ERModelRelationshipKeyClass
+    dataHubPolicyKey: DataHubPolicyKeyClass
+    dataHubRetentionKey: DataHubRetentionKeyClass
+    dataContractKey: DataContractKeyClass
     dataProcessKey: DataProcessKeyClass
+    dataProcessInstanceKey: DataProcessInstanceKeyClass
     glossaryTermKey: GlossaryTermKeyClass
-    mlFeatureTableKey: MLFeatureTableKeyClass
-    mlFeatureKey: MLFeatureKeyClass
-    dataHubRoleKey: DataHubRoleKeyClass
-    telemetryKey: TelemetryKeyClass
-    dataHubConnectionKey: DataHubConnectionKeyClass
-    dataJobInfo: DataJobInfoClass
-    dataJobInputOutput: DataJobInputOutputClass
-    versionInfo: VersionInfoClass
-    editableDataFlowProperties: EditableDataFlowPropertiesClass
-    dataFlowInfo: DataFlowInfoClass
-    editableDataJobProperties: EditableDataJobPropertiesClass
-    datahubIngestionCheckpoint: DatahubIngestionCheckpointClass
-    datahubIngestionRunSummary: DatahubIngestionRunSummaryClass
-    businessAttributeInfo: BusinessAttributeInfoClass
-    businessAttributeKey: BusinessAttributeKeyClass
-    businessAttributes: BusinessAttributesClass
-    dataHubSecretValue: DataHubSecretValueClass
-    glossaryTermInfo: GlossaryTermInfoClass
-    glossaryNodeInfo: GlossaryNodeInfoClass
-    glossaryRelatedTerms: GlossaryRelatedTermsClass
-    mlModelGroupProperties: MLModelGroupPropertiesClass
-    intendedUse: IntendedUseClass
-    mlModelEthicalConsiderations: EthicalConsiderationsClass
-    mlModelCaveatsAndRecommendations: CaveatsAndRecommendationsClass
-    mlMetric: MLMetricClass
-    mlHyperParam: MLHyperParamClass
-    mlTrainingRunProperties: MLTrainingRunPropertiesClass
-    editableMlFeatureProperties: EditableMLFeaturePropertiesClass
-    mlModelTrainingData: TrainingDataClass
-    mlModelQuantitativeAnalyses: QuantitativeAnalysesClass
-    mlModelDeploymentProperties: MLModelDeploymentPropertiesClass
-    mlModelMetrics: MetricsClass
-    editableMlPrimaryKeyProperties: EditableMLPrimaryKeyPropertiesClass
-    mlFeatureProperties: MLFeaturePropertiesClass
-    editableMlModelProperties: EditableMLModelPropertiesClass
-    mlModelProperties: MLModelPropertiesClass
-    editableMlFeatureTableProperties: EditableMLFeatureTablePropertiesClass
-    sourceCode: SourceCodeClass
-    editableMlModelGroupProperties: EditableMLModelGroupPropertiesClass
-    mlPrimaryKeyProperties: MLPrimaryKeyPropertiesClass
-    mlModelFactorPrompts: MLModelFactorPromptsClass
-    mlModelEvaluationData: EvaluationDataClass
-    mlFeatureTableProperties: MLFeatureTablePropertiesClass
-    dataHubViewInfo: DataHubViewInfoClass
-    telemetryClientId: TelemetryClientIdClass
-    dashboardUsageStatistics: DashboardUsageStatisticsClass
-    editableDashboardProperties: EditableDashboardPropertiesClass
-    dashboardInfo: DashboardInfoClass
-    dataProcessInstanceInput: DataProcessInstanceInputClass
+    formKey: FormKeyClass
+    schemaFieldKey: SchemaFieldKeyClass
+    tagKey: TagKeyClass
+    postKey: PostKeyClass
+    dataHubIngestionSourceKey: DataHubIngestionSourceKeyClass
+    inviteTokenKey: InviteTokenKeyClass
+    dataHubPersonaKey: DataHubPersonaKeyClass
+    containerKey: ContainerKeyClass
+    incidentKey: IncidentKeyClass
+    dataPlatformInstanceKey: DataPlatformInstanceKeyClass
+    testKey: TestKeyClass
+    versionSetKey: VersionSetKeyClass
+    dataHubSecretKey: DataHubSecretKeyClass
+    dataHubPersonaInfo: DataHubPersonaInfoClass
+    dataPlatformInstanceProperties: DataPlatformInstancePropertiesClass
+    icebergWarehouseInfo: IcebergWarehouseInfoClass
+    domains: DomainsClass
+    domainProperties: DomainPropertiesClass
+    ownershipTypeInfo: OwnershipTypeInfoClass
+    schemafieldInfo: SchemaFieldInfoClass
+    schemaFieldAliases: SchemaFieldAliasesClass
+    versionSetProperties: VersionSetPropertiesClass
+    testResults: TestResultsClass
+    testInfo: TestInfoClass
+    dataProcessInfo: DataProcessInfoClass
     dataProcessInstanceOutput: DataProcessInstanceOutputClass
     dataProcessInstanceRelationships: DataProcessInstanceRelationshipsClass
-    dataProcessInfo: DataProcessInfoClass
-    dataProcessInstanceRunEvent: DataProcessInstanceRunEventClass
     dataProcessInstanceProperties: DataProcessInstancePropertiesClass
-    dataTypeKey: DataTypeKeyClass
-    dataTypeInfo: DataTypeInfoClass
-    formInfo: FormInfoClass
-    dynamicFormAssignment: DynamicFormAssignmentClass
-    dataHubConnectionDetails: DataHubConnectionDetailsClass
-    dataProductKey: DataProductKeyClass
-    dataProductProperties: DataProductPropertiesClass
-    dataHubPersonaInfo: DataHubPersonaInfoClass
-    entityTypeInfo: EntityTypeInfoClass
+    dataProcessInstanceInput: DataProcessInstanceInputClass
+    dataProcessInstanceRunEvent: DataProcessInstanceRunEventClass
     entityTypeKey: EntityTypeKeyClass
-    postInfo: PostInfoClass
+    entityTypeInfo: EntityTypeInfoClass
+    editableERModelRelationshipProperties: EditableERModelRelationshipPropertiesClass
+    erModelRelationshipProperties: ERModelRelationshipPropertiesClass
+    corpUserStatus: CorpUserStatusClass
+    corpUserCredentials: CorpUserCredentialsClass
+    corpUserEditableInfo: CorpUserEditableInfoClass
+    corpGroupInfo: CorpGroupInfoClass
+    corpGroupEditableInfo: CorpGroupEditableInfoClass
+    corpUserInfo: CorpUserInfoClass
+    corpUserSettings: CorpUserSettingsClass
+    roleMembership: RoleMembershipClass
+    groupMembership: GroupMembershipClass
+    nativeGroupMembership: NativeGroupMembershipClass
+    inviteToken: InviteTokenClass
     roleProperties: RolePropertiesClass
     actors: ActorsClass
-    dataContractProperties: DataContractPropertiesClass
-    dataContractStatus: DataContractStatusClass
-    erModelRelationshipProperties: ERModelRelationshipPropertiesClass
-    editableERModelRelationshipProperties: EditableERModelRelationshipPropertiesClass
-    chartUsageStatistics: ChartUsageStatisticsClass
-    chartQuery: ChartQueryClass
-    chartInfo: ChartInfoClass
-    editableChartProperties: EditableChartPropertiesClass
-    queryProperties: QueryPropertiesClass
-    querySubjects: QuerySubjectsClass
-    queryUsageStatistics: QueryUsageStatisticsClass
-    versionSetProperties: VersionSetPropertiesClass
+    postInfo: PostInfoClass
+    editableDashboardProperties: EditableDashboardPropertiesClass
+    dashboardInfo: DashboardInfoClass
+    dashboardUsageStatistics: DashboardUsageStatisticsClass
+    notebookInfo: NotebookInfoClass
+    notebookContent: NotebookContentClass
+    editableNotebookProperties: EditableNotebookPropertiesClass
     dataHubIngestionSourceInfo: DataHubIngestionSourceInfoClass
-    tagProperties: TagPropertiesClass
-    domains: DomainsClass
-    domainProperties: DomainPropertiesClass
-    editableSchemaMetadata: EditableSchemaMetadataClass
-    schemaMetadata: SchemaMetadataClass
-    globalSettingsInfo: GlobalSettingsInfoClass
-    viewProperties: ViewPropertiesClass
-    datasetUpstreamLineage: DatasetUpstreamLineageClass
-    datasetProfile: DatasetProfileClass
-    editableDatasetProperties: EditableDatasetPropertiesClass
-    icebergCatalogInfo: IcebergCatalogInfoClass
-    datasetUsageStatistics: DatasetUsageStatisticsClass
-    partitionsSummary: PartitionsSummaryClass
-    datasetProperties: DatasetPropertiesClass
-    upstreamLineage: UpstreamLineageClass
-    datasetDeprecation: DatasetDeprecationClass
-    editableContainerProperties: EditableContainerPropertiesClass
-    container: ContainerClass
-    containerProperties: ContainerPropertiesClass
-    schemaFieldAliases: SchemaFieldAliasesClass
-    schemafieldInfo: SchemaFieldInfoClass
-    platformResourceKey: PlatformResourceKeyClass
-    platformResourceInfo: PlatformResourceInfoClass
-    testInfo: TestInfoClass
-    testResults: TestResultsClass
     structuredProperties: StructuredPropertiesClass
     propertyDefinition: StructuredPropertyDefinitionClass
     structuredPropertyKey: StructuredPropertyKeyClass
     structuredPropertySettings: StructuredPropertySettingsClass
+    tagProperties: TagPropertiesClass
+    mlModelEthicalConsiderations: EthicalConsiderationsClass
+    mlModelDeploymentProperties: MLModelDeploymentPropertiesClass
+    mlPrimaryKeyProperties: MLPrimaryKeyPropertiesClass
+    mlModelTrainingData: TrainingDataClass
+    mlModelFactorPrompts: MLModelFactorPromptsClass
+    mlModelQuantitativeAnalyses: QuantitativeAnalysesClass
+    mlHyperParam: MLHyperParamClass
+    editableMlFeatureTableProperties: EditableMLFeatureTablePropertiesClass
+    editableMlPrimaryKeyProperties: EditableMLPrimaryKeyPropertiesClass
+    sourceCode: SourceCodeClass
+    mlModelEvaluationData: EvaluationDataClass
+    mlTrainingRunProperties: MLTrainingRunPropertiesClass
+    editableMlModelGroupProperties: EditableMLModelGroupPropertiesClass
+    mlFeatureTableProperties: MLFeatureTablePropertiesClass
+    mlModelMetrics: MetricsClass
+    mlMetric: MLMetricClass
+    mlModelCaveatsAndRecommendations: CaveatsAndRecommendationsClass
+    mlModelGroupProperties: MLModelGroupPropertiesClass
+    mlFeatureProperties: MLFeaturePropertiesClass
+    intendedUse: IntendedUseClass
+    editableMlFeatureProperties: EditableMLFeaturePropertiesClass
+    mlModelProperties: MLModelPropertiesClass
+    editableMlModelProperties: EditableMLModelPropertiesClass
+    platformResourceKey: PlatformResourceKeyClass
+    platformResourceInfo: PlatformResourceInfoClass
+    chartUsageStatistics: ChartUsageStatisticsClass
+    chartQuery: ChartQueryClass
+    editableChartProperties: EditableChartPropertiesClass
+    chartInfo: ChartInfoClass
+    dataHubRetentionConfig: DataHubRetentionConfigClass
 
 
 KEY_ASPECTS: Dict[str, Type[_Aspect]] = {
-    'erModelRelationship': ERModelRelationshipKeyClass,
-    'dataProcessInstance': DataProcessInstanceKeyClass,
-    'chart': ChartKeyClass,
-    'versionSet': VersionSetKeyClass,
-    'dataHubUpgrade': DataHubUpgradeKeyClass,
+    'businessAttribute': BusinessAttributeKeyClass,
+    'dataType': DataTypeKeyClass,
+    'dataProduct': DataProductKeyClass,
+    'telemetry': TelemetryKeyClass,
     'dataHubAccessToken': DataHubAccessTokenKeyClass,
-    'mlPrimaryKey': MLPrimaryKeyKeyClass,
-    'ownershipType': OwnershipTypeKeyClass,
-    'schemaField': SchemaFieldKeyClass,
-    'post': PostKeyClass,
-    'dataHubSecret': DataHubSecretKeyClass,
-    'dashboard': DashboardKeyClass,
-    'mlModelGroup': MLModelGroupKeyClass,
-    'dataset': DatasetKeyClass,
+    'dataHubConnection': DataHubConnectionKeyClass,
+    'dataHubRole': DataHubRoleKeyClass,
+    'mlModel': MLModelKeyClass,
     'notebook': NotebookKeyClass,
-    'dataFlow': DataFlowKeyClass,
-    'mlModelDeployment': MLModelDeploymentKeyClass,
-    'form': FormKeyClass,
-    'domain': DomainKeyClass,
-    'incident': IncidentKeyClass,
-    'tag': TagKeyClass,
     'role': RoleKeyClass,
-    'dataPlatform': DataPlatformKeyClass,
-    'dataJob': DataJobKeyClass,
-    'dataHubStepState': DataHubStepStateKeyClass,
-    'dataHubView': DataHubViewKeyClass,
-    'corpuser': CorpUserKeyClass,
-    'dataHubRetention': DataHubRetentionKeyClass,
-    'dataPlatformInstance': DataPlatformInstanceKeyClass,
-    'dataHubPersona': DataHubPersonaKeyClass,
-    'container': ContainerKeyClass,
-    'glossaryNode': GlossaryNodeKeyClass,
-    'dataHubIngestionSource': DataHubIngestionSourceKeyClass,
-    'test': TestKeyClass,
-    'dataHubPolicy': DataHubPolicyKeyClass,
-    'corpGroup': CorpGroupKeyClass,
-    'dataContract': DataContractKeyClass,
     'globalSettings': GlobalSettingsKeyClass,
+    'dataset': DatasetKeyClass,
+    'chart': ChartKeyClass,
+    'glossaryNode': GlossaryNodeKeyClass,
     'assertion': AssertionKeyClass,
+    'dataHubExecutionRequest': ExecutionRequestKeyClass,
+    'mlModelGroup': MLModelGroupKeyClass,
+    'mlModelDeployment': MLModelDeploymentKeyClass,
+    'dataFlow': DataFlowKeyClass,
+    'dataJob': DataJobKeyClass,
+    'corpGroup': CorpGroupKeyClass,
+    'dataPlatform': DataPlatformKeyClass,
+    'corpuser': CorpUserKeyClass,
+    'dashboard': DashboardKeyClass,
+    'mlPrimaryKey': MLPrimaryKeyKeyClass,
     'query': QueryKeyClass,
+    'ownershipType': OwnershipTypeKeyClass,
+    'domain': DomainKeyClass,
+    'mlFeature': MLFeatureKeyClass,
+    'mlFeatureTable': MLFeatureTableKeyClass,
+    'dataHubView': DataHubViewKeyClass,
     'dataHubAction': DataHubActionKeyClass,
-    'mlModel': MLModelKeyClass,
-    'dataHubExecutionRequest': ExecutionRequestKeyClass,
-    'inviteToken': InviteTokenKeyClass,
+    'dataHubUpgrade': DataHubUpgradeKeyClass,
+    'dataHubStepState': DataHubStepStateKeyClass,
+    'erModelRelationship': ERModelRelationshipKeyClass,
+    'dataHubPolicy': DataHubPolicyKeyClass,
+    'dataHubRetention': DataHubRetentionKeyClass,
+    'dataContract': DataContractKeyClass,
     'dataProcess': DataProcessKeyClass,
+    'dataProcessInstance': DataProcessInstanceKeyClass,
     'glossaryTerm': GlossaryTermKeyClass,
-    'mlFeatureTable': MLFeatureTableKeyClass,
-    'mlFeature': MLFeatureKeyClass,
-    'dataHubRole': DataHubRoleKeyClass,
-    'telemetry': TelemetryKeyClass,
-    'dataHubConnection': DataHubConnectionKeyClass,
-    'businessAttribute': BusinessAttributeKeyClass,
-    'dataType': DataTypeKeyClass,
-    'dataProduct': DataProductKeyClass,
+    'form': FormKeyClass,
+    'schemaField': SchemaFieldKeyClass,
+    'tag': TagKeyClass,
+    'post': PostKeyClass,
+    'dataHubIngestionSource': DataHubIngestionSourceKeyClass,
+    'inviteToken': InviteTokenKeyClass,
+    'dataHubPersona': DataHubPersonaKeyClass,
+    'container': ContainerKeyClass,
+    'incident': IncidentKeyClass,
+    'dataPlatformInstance': DataPlatformInstanceKeyClass,
+    'test': TestKeyClass,
+    'versionSet': VersionSetKeyClass,
+    'dataHubSecret': DataHubSecretKeyClass,
     'entityType': EntityTypeKeyClass,
-    'platformResource': PlatformResourceKeyClass,
-    'structuredProperty': StructuredPropertyKeyClass
+    'structuredProperty': StructuredPropertyKeyClass,
+    'platformResource': PlatformResourceKeyClass
 }
 
 # fmt: on