acryl-datahub-cloud 0.3.7.9rc1__py3-none-any.whl → 0.3.7.9.1__py3-none-any.whl

acryl_datahub_cloud/metadata/schema_classes.py

@@ -13056,7 +13056,7 @@ class DataProcessInstanceInputClass(_Aspect):
     
     @property
     def inputs(self) -> List[str]:
-        """Input datasets to be consumed"""
+        """Input assets consumed"""
         return self._inner_dict.get('inputs')  # type: ignore
     
     @inputs.setter
@@ -21313,7 +21313,7 @@ class DataProcessInstanceKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'dataProcessInstanceKey'
-    ASPECT_INFO = {'keyForEntity': 'dataProcessInstance', 'entityCategory': '_unset_', 'entityAspects': ['dataProcessInstanceInput', 'dataProcessInstanceOutput', 'dataProcessInstanceProperties', 'dataProcessInstanceRelationships', 'dataProcessInstanceRunEvent', 'status', 'testResults', 'lineageFeatures'], 'entityDoc': 'DataProcessInstance represents an instance of a datajob/jobflow run'}
+    ASPECT_INFO = {'keyForEntity': 'dataProcessInstance', 'entityCategory': '_unset_', 'entityAspects': ['dataProcessInstanceInput', 'dataProcessInstanceOutput', 'dataProcessInstanceProperties', 'dataProcessInstanceRelationships', 'dataProcessInstanceRunEvent', 'status', 'testResults', 'dataPlatformInstance', 'subTypes', 'container', 'mlTrainingRunProperties', 'lineageFeatures'], 'entityDoc': 'DataProcessInstance represents an instance of a datajob/jobflow run'}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.DataProcessInstanceKey")
 
     def __init__(self,
@@ -25298,8 +25298,13 @@ class MLModelGroupPropertiesClass(_Aspect):
 
     def __init__(self,
         customProperties: Optional[Dict[str, str]]=None,
+        trainingJobs: Union[None, List[str]]=None,
+        downstreamJobs: Union[None, List[str]]=None,
+        name: Union[None, str]=None,
         description: Union[None, str]=None,
         createdAt: Union[None, int]=None,
+        created: Union[None, "TimeStampClass"]=None,
+        lastModified: Union[None, "TimeStampClass"]=None,
         version: Union[None, "VersionTagClass"]=None,
     ):
         super().__init__()
@@ -25309,14 +25314,24 @@ class MLModelGroupPropertiesClass(_Aspect):
             self.customProperties = dict()
         else:
             self.customProperties = customProperties
+        self.trainingJobs = trainingJobs
+        self.downstreamJobs = downstreamJobs
+        self.name = name
         self.description = description
         self.createdAt = createdAt
+        self.created = created
+        self.lastModified = lastModified
         self.version = version
     
     def _restore_defaults(self) -> None:
         self.customProperties = dict()
+        self.trainingJobs = self.RECORD_SCHEMA.fields_dict["trainingJobs"].default
+        self.downstreamJobs = self.RECORD_SCHEMA.fields_dict["downstreamJobs"].default
+        self.name = self.RECORD_SCHEMA.fields_dict["name"].default
         self.description = self.RECORD_SCHEMA.fields_dict["description"].default
         self.createdAt = self.RECORD_SCHEMA.fields_dict["createdAt"].default
+        self.created = self.RECORD_SCHEMA.fields_dict["created"].default
+        self.lastModified = self.RECORD_SCHEMA.fields_dict["lastModified"].default
         self.version = self.RECORD_SCHEMA.fields_dict["version"].default
     
     
@@ -25330,6 +25345,36 @@ class MLModelGroupPropertiesClass(_Aspect):
         self._inner_dict['customProperties'] = value
     
     
+    @property
+    def trainingJobs(self) -> Union[None, List[str]]:
+        """List of jobs or process instances (if any) used to train the model or group. Visible in Lineage. Note that ML Models can also be specified as the output of a specific Data Process Instances (runs) via the DataProcessInstanceOutputs aspect."""
+        return self._inner_dict.get('trainingJobs')  # type: ignore
+    
+    @trainingJobs.setter
+    def trainingJobs(self, value: Union[None, List[str]]) -> None:
+        self._inner_dict['trainingJobs'] = value
+    
+    
+    @property
+    def downstreamJobs(self) -> Union[None, List[str]]:
+        """List of jobs or process instances (if any) that use the model or group."""
+        return self._inner_dict.get('downstreamJobs')  # type: ignore
+    
+    @downstreamJobs.setter
+    def downstreamJobs(self, value: Union[None, List[str]]) -> None:
+        self._inner_dict['downstreamJobs'] = value
+    
+    
+    @property
+    def name(self) -> Union[None, str]:
+        """Display name of the MLModelGroup"""
+        return self._inner_dict.get('name')  # type: ignore
+    
+    @name.setter
+    def name(self, value: Union[None, str]) -> None:
+        self._inner_dict['name'] = value
+    
+    
     @property
     def description(self) -> Union[None, str]:
         """Documentation of the MLModelGroup"""
@@ -25350,6 +25395,26 @@ class MLModelGroupPropertiesClass(_Aspect):
         self._inner_dict['createdAt'] = value
     
     
+    @property
+    def created(self) -> Union[None, "TimeStampClass"]:
+        """Time and Actor who created the MLModelGroup"""
+        return self._inner_dict.get('created')  # type: ignore
+    
+    @created.setter
+    def created(self, value: Union[None, "TimeStampClass"]) -> None:
+        self._inner_dict['created'] = value
+    
+    
+    @property
+    def lastModified(self) -> Union[None, "TimeStampClass"]:
+        """Date when the MLModelGroup was last modified"""
+        return self._inner_dict.get('lastModified')  # type: ignore
+    
+    @lastModified.setter
+    def lastModified(self, value: Union[None, "TimeStampClass"]) -> None:
+        self._inner_dict['lastModified'] = value
+    
+    
     @property
     def version(self) -> Union[None, "VersionTagClass"]:
         """Version of the MLModelGroup"""
@@ -25371,8 +25436,13 @@ class MLModelPropertiesClass(_Aspect):
     def __init__(self,
         customProperties: Optional[Dict[str, str]]=None,
         externalUrl: Union[None, str]=None,
+        trainingJobs: Union[None, List[str]]=None,
+        downstreamJobs: Union[None, List[str]]=None,
+        name: Union[None, str]=None,
         description: Union[None, str]=None,
         date: Union[None, int]=None,
+        created: Union[None, "TimeStampClass"]=None,
+        lastModified: Union[None, "TimeStampClass"]=None,
         version: Union[None, "VersionTagClass"]=None,
         type: Union[None, str]=None,
         hyperParameters: Union[None, Dict[str, Union[str, int, float, float, bool]]]=None,
@@ -25382,8 +25452,6 @@ class MLModelPropertiesClass(_Aspect):
         mlFeatures: Union[None, List[str]]=None,
         tags: Optional[List[str]]=None,
         deployments: Union[None, List[str]]=None,
-        trainingJobs: Union[None, List[str]]=None,
-        downstreamJobs: Union[None, List[str]]=None,
         groups: Union[None, List[str]]=None,
     ):
         super().__init__()
@@ -25394,8 +25462,13 @@ class MLModelPropertiesClass(_Aspect):
         else:
             self.customProperties = customProperties
         self.externalUrl = externalUrl
+        self.trainingJobs = trainingJobs
+        self.downstreamJobs = downstreamJobs
+        self.name = name
         self.description = description
         self.date = date
+        self.created = created
+        self.lastModified = lastModified
         self.version = version
         self.type = type
         self.hyperParameters = hyperParameters
@@ -25409,15 +25482,18 @@ class MLModelPropertiesClass(_Aspect):
         else:
             self.tags = tags
         self.deployments = deployments
-        self.trainingJobs = trainingJobs
-        self.downstreamJobs = downstreamJobs
         self.groups = groups
     
     def _restore_defaults(self) -> None:
         self.customProperties = dict()
         self.externalUrl = self.RECORD_SCHEMA.fields_dict["externalUrl"].default
+        self.trainingJobs = self.RECORD_SCHEMA.fields_dict["trainingJobs"].default
+        self.downstreamJobs = self.RECORD_SCHEMA.fields_dict["downstreamJobs"].default
+        self.name = self.RECORD_SCHEMA.fields_dict["name"].default
         self.description = self.RECORD_SCHEMA.fields_dict["description"].default
         self.date = self.RECORD_SCHEMA.fields_dict["date"].default
+        self.created = self.RECORD_SCHEMA.fields_dict["created"].default
+        self.lastModified = self.RECORD_SCHEMA.fields_dict["lastModified"].default
         self.version = self.RECORD_SCHEMA.fields_dict["version"].default
         self.type = self.RECORD_SCHEMA.fields_dict["type"].default
         self.hyperParameters = self.RECORD_SCHEMA.fields_dict["hyperParameters"].default
@@ -25427,8 +25503,6 @@ class MLModelPropertiesClass(_Aspect):
         self.mlFeatures = self.RECORD_SCHEMA.fields_dict["mlFeatures"].default
         self.tags = list()
         self.deployments = self.RECORD_SCHEMA.fields_dict["deployments"].default
-        self.trainingJobs = self.RECORD_SCHEMA.fields_dict["trainingJobs"].default
-        self.downstreamJobs = self.RECORD_SCHEMA.fields_dict["downstreamJobs"].default
         self.groups = self.RECORD_SCHEMA.fields_dict["groups"].default
     
     
@@ -25452,6 +25526,36 @@ class MLModelPropertiesClass(_Aspect):
         self._inner_dict['externalUrl'] = value
     
     
+    @property
+    def trainingJobs(self) -> Union[None, List[str]]:
+        """List of jobs or process instances (if any) used to train the model or group. Visible in Lineage. Note that ML Models can also be specified as the output of a specific Data Process Instances (runs) via the DataProcessInstanceOutputs aspect."""
+        return self._inner_dict.get('trainingJobs')  # type: ignore
+    
+    @trainingJobs.setter
+    def trainingJobs(self, value: Union[None, List[str]]) -> None:
+        self._inner_dict['trainingJobs'] = value
+    
+    
+    @property
+    def downstreamJobs(self) -> Union[None, List[str]]:
+        """List of jobs or process instances (if any) that use the model or group."""
+        return self._inner_dict.get('downstreamJobs')  # type: ignore
+    
+    @downstreamJobs.setter
+    def downstreamJobs(self, value: Union[None, List[str]]) -> None:
+        self._inner_dict['downstreamJobs'] = value
+    
+    
+    @property
+    def name(self) -> Union[None, str]:
+        """Display name of the MLModel"""
+        return self._inner_dict.get('name')  # type: ignore
+    
+    @name.setter
+    def name(self, value: Union[None, str]) -> None:
+        self._inner_dict['name'] = value
+    
+    
     @property
     def description(self) -> Union[None, str]:
         """Documentation of the MLModel"""
@@ -25472,6 +25576,26 @@ class MLModelPropertiesClass(_Aspect):
         self._inner_dict['date'] = value
     
     
+    @property
+    def created(self) -> Union[None, "TimeStampClass"]:
+        """Audit stamp containing who created this and when"""
+        return self._inner_dict.get('created')  # type: ignore
+    
+    @created.setter
+    def created(self, value: Union[None, "TimeStampClass"]) -> None:
+        self._inner_dict['created'] = value
+    
+    
+    @property
+    def lastModified(self) -> Union[None, "TimeStampClass"]:
+        """Date when the MLModel was last modified"""
+        return self._inner_dict.get('lastModified')  # type: ignore
+    
+    @lastModified.setter
+    def lastModified(self, value: Union[None, "TimeStampClass"]) -> None:
+        self._inner_dict['lastModified'] = value
+    
+    
     @property
     def version(self) -> Union[None, "VersionTagClass"]:
         """Version of the MLModel"""
@@ -25564,26 +25688,6 @@ class MLModelPropertiesClass(_Aspect):
         self._inner_dict['deployments'] = value
     
     
-    @property
-    def trainingJobs(self) -> Union[None, List[str]]:
-        """List of jobs (if any) used to train the model"""
-        return self._inner_dict.get('trainingJobs')  # type: ignore
-    
-    @trainingJobs.setter
-    def trainingJobs(self, value: Union[None, List[str]]) -> None:
-        self._inner_dict['trainingJobs'] = value
-    
-    
-    @property
-    def downstreamJobs(self) -> Union[None, List[str]]:
-        """List of jobs (if any) that use the model"""
-        return self._inner_dict.get('downstreamJobs')  # type: ignore
-    
-    @downstreamJobs.setter
-    def downstreamJobs(self, value: Union[None, List[str]]) -> None:
-        self._inner_dict['downstreamJobs'] = value
-    
-    
     @property
     def groups(self) -> Union[None, List[str]]:
         """Groups the model belongs to"""
@@ -25679,6 +25783,104 @@ class MLPrimaryKeyPropertiesClass(_Aspect):
         self._inner_dict['sources'] = value
     
     
+class MLTrainingRunPropertiesClass(_Aspect):
+    """The inputs and outputs of this training run"""
+
+
+    ASPECT_NAME = 'mlTrainingRunProperties'
+    ASPECT_INFO = {}
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.ml.metadata.MLTrainingRunProperties")
+
+    def __init__(self,
+        customProperties: Optional[Dict[str, str]]=None,
+        externalUrl: Union[None, str]=None,
+        id: Union[None, str]=None,
+        outputUrls: Union[None, List[str]]=None,
+        hyperParams: Union[None, List["MLHyperParamClass"]]=None,
+        trainingMetrics: Union[None, List["MLMetricClass"]]=None,
+    ):
+        super().__init__()
+        
+        if customProperties is None:
+            # default: {}
+            self.customProperties = dict()
+        else:
+            self.customProperties = customProperties
+        self.externalUrl = externalUrl
+        self.id = id
+        self.outputUrls = outputUrls
+        self.hyperParams = hyperParams
+        self.trainingMetrics = trainingMetrics
+    
+    def _restore_defaults(self) -> None:
+        self.customProperties = dict()
+        self.externalUrl = self.RECORD_SCHEMA.fields_dict["externalUrl"].default
+        self.id = self.RECORD_SCHEMA.fields_dict["id"].default
+        self.outputUrls = self.RECORD_SCHEMA.fields_dict["outputUrls"].default
+        self.hyperParams = self.RECORD_SCHEMA.fields_dict["hyperParams"].default
+        self.trainingMetrics = self.RECORD_SCHEMA.fields_dict["trainingMetrics"].default
+    
+    
+    @property
+    def customProperties(self) -> Dict[str, str]:
+        """Custom property bag."""
+        return self._inner_dict.get('customProperties')  # type: ignore
+    
+    @customProperties.setter
+    def customProperties(self, value: Dict[str, str]) -> None:
+        self._inner_dict['customProperties'] = value
+    
+    
+    @property
+    def externalUrl(self) -> Union[None, str]:
+        """URL where the reference exist"""
+        return self._inner_dict.get('externalUrl')  # type: ignore
+    
+    @externalUrl.setter
+    def externalUrl(self, value: Union[None, str]) -> None:
+        self._inner_dict['externalUrl'] = value
+    
+    
+    @property
+    def id(self) -> Union[None, str]:
+        """Run Id of the ML Training Run"""
+        return self._inner_dict.get('id')  # type: ignore
+    
+    @id.setter
+    def id(self, value: Union[None, str]) -> None:
+        self._inner_dict['id'] = value
+    
+    
+    @property
+    def outputUrls(self) -> Union[None, List[str]]:
+        """List of URLs for the Outputs of the ML Training Run"""
+        return self._inner_dict.get('outputUrls')  # type: ignore
+    
+    @outputUrls.setter
+    def outputUrls(self, value: Union[None, List[str]]) -> None:
+        self._inner_dict['outputUrls'] = value
+    
+    
+    @property
+    def hyperParams(self) -> Union[None, List["MLHyperParamClass"]]:
+        """Hyperparameters of the ML Training Run"""
+        return self._inner_dict.get('hyperParams')  # type: ignore
+    
+    @hyperParams.setter
+    def hyperParams(self, value: Union[None, List["MLHyperParamClass"]]) -> None:
+        self._inner_dict['hyperParams'] = value
+    
+    
+    @property
+    def trainingMetrics(self) -> Union[None, List["MLMetricClass"]]:
+        """Metrics of the ML Training Run"""
+        return self._inner_dict.get('trainingMetrics')  # type: ignore
+    
+    @trainingMetrics.setter
+    def trainingMetrics(self, value: Union[None, List["MLMetricClass"]]) -> None:
+        self._inner_dict['trainingMetrics'] = value
+    
+    
 class MetricsClass(_Aspect):
     """Metrics to be featured for the MLModel."""
 
@@ -34812,6 +35014,7 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.ml.metadata.MLModelGroupProperties': MLModelGroupPropertiesClass,
     'com.linkedin.pegasus2avro.ml.metadata.MLModelProperties': MLModelPropertiesClass,
     'com.linkedin.pegasus2avro.ml.metadata.MLPrimaryKeyProperties': MLPrimaryKeyPropertiesClass,
+    'com.linkedin.pegasus2avro.ml.metadata.MLTrainingRunProperties': MLTrainingRunPropertiesClass,
     'com.linkedin.pegasus2avro.ml.metadata.Metrics': MetricsClass,
     'com.linkedin.pegasus2avro.ml.metadata.QuantitativeAnalyses': QuantitativeAnalysesClass,
     'com.linkedin.pegasus2avro.ml.metadata.SourceCode': SourceCodeClass,
@@ -35472,6 +35675,7 @@ __SCHEMA_TYPES = {
     'MLModelGroupProperties': MLModelGroupPropertiesClass,
     'MLModelProperties': MLModelPropertiesClass,
     'MLPrimaryKeyProperties': MLPrimaryKeyPropertiesClass,
+    'MLTrainingRunProperties': MLTrainingRunPropertiesClass,
     'Metrics': MetricsClass,
     'QuantitativeAnalyses': QuantitativeAnalysesClass,
     'SourceCode': SourceCodeClass,
@@ -35660,256 +35864,257 @@ avrojson.set_global_json_converter(_json_converter)
     
 
 ASPECT_CLASSES: List[Type[_Aspect]] = [
-    TagPropertiesClass,
-    StatusClass,
-    IncidentsSummaryClass,
+    DataHubRetentionConfigClass,
+    IncidentInfoClass,
+    IncidentActivityEventClass,
+    IncidentNotificationDetailsClass,
+    IncidentSourceClass,
+    DataHubConnectionDetailsClass,
+    DeprecationClass,
+    ShareClass,
+    BrowsePathsV2Class,
+    OriginClass,
+    GlossaryTermsClass,
     CostClass,
-    BrowsePathsClass,
+    AccessClass,
+    SubTypesClass,
+    SiblingsClass,
+    IncidentsSummaryClass,
+    InstitutionalMemoryClass,
     AssertionsSummaryClass,
+    BrowsePathsClass,
+    StatusClass,
+    DocumentationClass,
+    DataPlatformInstanceClass,
     AnomaliesSummaryClass,
-    InstitutionalMemoryClass,
+    InputFieldsClass,
+    DisplayPropertiesClass,
     ProposalsClass,
-    SubTypesClass,
+    OwnershipClass,
+    OperationClass,
     FormsClass,
-    DeprecationClass,
-    GlobalTagsClass,
     EmbedClass,
-    OperationClass,
-    DataPlatformInstanceClass,
-    GlossaryTermsClass,
-    SiblingsClass,
-    InputFieldsClass,
-    DisplayPropertiesClass,
-    OwnershipClass,
-    DocumentationClass,
-    BrowsePathsV2Class,
-    OriginClass,
-    ShareClass,
-    AccessClass,
-    DataHubAccessTokenInfoClass,
+    GlobalTagsClass,
     GlobalSettingsInfoClass,
-    AiInferenceMetadataClass,
-    EntityInferenceMetadataClass,
-    SchemaMetadataClass,
-    SchemaProposalsClass,
-    EditableSchemaMetadataClass,
-    InferredNeighborsClass,
-    InferredMetadataClass,
-    SchemaFieldsInferredNeighborsClass,
-    SchemaFieldsInferredMetadataClass,
-    DataPlatformInstancePropertiesClass,
-    ERModelRelationshipPropertiesClass,
-    EditableERModelRelationshipPropertiesClass,
+    ConstraintInfoClass,
+    StructuredPropertiesClass,
+    StructuredPropertyKeyClass,
     StructuredPropertyDefinitionClass,
     StructuredPropertySettingsClass,
-    StructuredPropertyKeyClass,
-    StructuredPropertiesClass,
-    RolePropertiesClass,
-    ActorsClass,
-    DashboardUsageStatisticsClass,
+    SchemaFieldInfoClass,
+    SchemaFieldAliasesClass,
+    SchemaFieldsInferredMetadataClass,
+    SchemaFieldsInferredNeighborsClass,
+    InferredMetadataClass,
+    InferredNeighborsClass,
+    ContainerPropertiesClass,
+    EditableContainerPropertiesClass,
+    ContainerClass,
+    DataHubUpgradeResultClass,
+    DataHubUpgradeRequestClass,
+    AnomalyInfoClass,
+    BusinessAttributesClass,
+    BusinessAttributeInfoClass,
+    BusinessAttributeKeyClass,
+    DataHubIngestionSourceInfoClass,
+    SubscriptionInfoClass,
+    TagPropertiesClass,
     EditableDashboardPropertiesClass,
+    DashboardUsageStatisticsClass,
     DashboardInfoClass,
-    OwnershipTypeInfoClass,
-    MonitorTimeseriesStateClass,
+    DataHubViewInfoClass,
+    DataContractStatusClass,
+    DataContractPropertiesClass,
+    RolePropertiesClass,
+    ActorsClass,
+    SchemaMetadataClass,
+    SchemaProposalsClass,
+    EditableSchemaMetadataClass,
+    DataPlatformInfoClass,
+    SlackUserInfoClass,
     MonitorInfoClass,
+    MonitorTimeseriesStateClass,
+    DataHubPersonaInfoClass,
+    DataHubPolicyInfoClass,
+    DataHubRoleInfoClass,
+    DataFlowInfoClass,
+    DataJobInfoClass,
+    VersionInfoClass,
+    EditableDataJobPropertiesClass,
+    EditableDataFlowPropertiesClass,
+    DataJobInputOutputClass,
+    DatahubIngestionRunSummaryClass,
+    DatahubIngestionCheckpointClass,
+    AssertionSummaryClass,
+    AssertionInfoClass,
+    AssertionAnalyticsRunEventClass,
+    AssertionDryRunEventClass,
+    AssertionInferenceDetailsClass,
+    AssertionRunEventClass,
+    AssertionActionsClass,
     GlossaryRelatedTermsClass,
     GlossaryNodeInfoClass,
     GlossaryTermInfoClass,
-    DataHubPersonaInfoClass,
-    SchemaFieldAliasesClass,
-    SchemaFieldInfoClass,
+    DomainsClass,
+    DomainPropertiesClass,
     ActionRequestInfoClass,
-    ActionRequestArchivedClass,
     ActionRequestStatusClass,
-    DataHubIngestionSourceInfoClass,
-    DataHubRetentionConfigClass,
-    DataHubActionInfoClass,
-    DataHubActionStatusClass,
-    DataPlatformInfoClass,
-    SlackUserInfoClass,
-    PostInfoClass,
-    EditableChartPropertiesClass,
-    ChartInfoClass,
-    ChartUsageStatisticsClass,
-    ChartQueryClass,
-    FormInfoClass,
-    DynamicFormAssignmentClass,
-    QueryUsageFeaturesClass,
-    QueryPropertiesClass,
-    QuerySubjectsClass,
-    QueryUsageStatisticsClass,
-    ExecutionRequestResultClass,
-    ExecutionRequestSignalClass,
-    ExecutionRequestInputClass,
-    DataTypeKeyClass,
-    DataTypeInfoClass,
-    DataProcessInstanceInputClass,
-    DataProcessInstanceRunEventClass,
-    DataProcessInstanceOutputClass,
-    DataProcessInstancePropertiesClass,
-    DataProcessInstanceRelationshipsClass,
-    DataProcessInfoClass,
-    DataHubRoleInfoClass,
-    DataHubPolicyInfoClass,
+    ActionRequestArchivedClass,
+    DataHubStepStatePropertiesClass,
+    CorpUserSettingsClass,
+    CorpUserEditableInfoClass,
+    NativeGroupMembershipClass,
+    CorpUserStatusClass,
+    CorpGroupInfoClass,
+    RoleMembershipClass,
+    InviteTokenClass,
+    CorpGroupSettingsClass,
+    GroupMembershipClass,
+    CorpUserInfoClass,
+    CorpUserCredentialsClass,
+    CorpGroupEditableInfoClass,
+    TelemetryClientIdClass,
+    DataProductKeyClass,
+    DataProductPropertiesClass,
+    NotebookInfoClass,
+    NotebookContentClass,
+    EditableNotebookPropertiesClass,
+    EditableMLFeatureTablePropertiesClass,
+    MLHyperParamClass,
+    MLFeaturePropertiesClass,
+    MLPrimaryKeyPropertiesClass,
+    MLModelPropertiesClass,
+    MLTrainingRunPropertiesClass,
+    EditableMLModelGroupPropertiesClass,
     MLModelDeploymentPropertiesClass,
-    MLFeatureTablePropertiesClass,
-    TrainingDataClass,
-    IntendedUseClass,
     EditableMLFeaturePropertiesClass,
-    EvaluationDataClass,
-    MLMetricClass,
-    MLFeaturePropertiesClass,
-    EditableMLFeatureTablePropertiesClass,
-    MetricsClass,
-    EditableMLPrimaryKeyPropertiesClass,
-    SourceCodeClass,
     EthicalConsiderationsClass,
-    MLPrimaryKeyPropertiesClass,
+    QuantitativeAnalysesClass,
+    SourceCodeClass,
+    EditableMLPrimaryKeyPropertiesClass,
     MLModelFactorPromptsClass,
-    EditableMLModelGroupPropertiesClass,
-    MLModelPropertiesClass,
+    MLMetricClass,
+    MetricsClass,
+    TrainingDataClass,
+    IntendedUseClass,
+    EvaluationDataClass,
+    MLFeatureTablePropertiesClass,
     CaveatsAndRecommendationsClass,
-    MLHyperParamClass,
-    QuantitativeAnalysesClass,
-    MLModelGroupPropertiesClass,
     EditableMLModelPropertiesClass,
-    IncidentNotificationDetailsClass,
-    IncidentActivityEventClass,
-    IncidentSourceClass,
-    IncidentInfoClass,
+    MLModelGroupPropertiesClass,
+    QueryPropertiesClass,
+    QueryUsageFeaturesClass,
+    QuerySubjectsClass,
+    QueryUsageStatisticsClass,
     LinkPreviewInfoClass,
-    AnomalyInfoClass,
-    DataHubConnectionDetailsClass,
-    DataContractPropertiesClass,
-    DataContractStatusClass,
-    DataHubStepStatePropertiesClass,
-    AssertionDryRunEventClass,
-    AssertionRunEventClass,
-    AssertionSummaryClass,
-    AssertionInfoClass,
-    AssertionInferenceDetailsClass,
-    AssertionAnalyticsRunEventClass,
-    AssertionActionsClass,
-    TelemetryClientIdClass,
     DataHubSecretValueClass,
-    DataHubUpgradeRequestClass,
-    DataHubUpgradeResultClass,
-    CorpGroupKeyClass,
-    GlobalSettingsKeyClass,
-    SchemaFieldKeyClass,
-    DataHubStepStateKeyClass,
-    CorpUserKeyClass,
-    DataHubConnectionKeyClass,
-    DashboardKeyClass,
+    DatasetUpstreamLineageClass,
+    UpstreamLineageClass,
+    DatasetProfileClass,
+    SchemaFieldProfileClass,
+    PartitionsSummaryClass,
+    DatasetPropertiesClass,
+    DatasetDeprecationClass,
+    EditableDatasetPropertiesClass,
+    DatasetUsageStatisticsClass,
+    ViewPropertiesClass,
+    FormInfoClass,
+    DynamicFormAssignmentClass,
+    DataTypeInfoClass,
+    DataTypeKeyClass,
+    TestResultsClass,
+    BatchTestRunEventClass,
+    TestInfoClass,
+    PlatformResourceInfoClass,
+    PlatformResourceKeyClass,
+    PostInfoClass,
+    DataProcessInstancePropertiesClass,
+    DataProcessInfoClass,
+    DataProcessInstanceRelationshipsClass,
+    DataProcessInstanceRunEventClass,
+    DataProcessInstanceInputClass,
+    DataProcessInstanceOutputClass,
+    EditableChartPropertiesClass,
+    ChartInfoClass,
+    ChartQueryClass,
+    ChartUsageStatisticsClass,
+    EditableERModelRelationshipPropertiesClass,
+    ERModelRelationshipPropertiesClass,
+    DataHubActionInfoClass,
+    DataHubActionStatusClass,
+    ExecutionRequestResultClass,
+    ExecutionRequestSignalClass,
+    ExecutionRequestInputClass,
+    RecommendationModuleClass,
+    MLModelGroupKeyClass,
+    DataHubPolicyKeyClass,
+    TagKeyClass,
+    MLModelDeploymentKeyClass,
+    DataHubActionKeyClass,
     DataHubRoleKeyClass,
-    ExecutionRequestKeyClass,
-    DataProcessInstanceKeyClass,
+    DataHubUpgradeKeyClass,
+    FormKeyClass,
+    NotebookKeyClass,
+    DataHubConnectionKeyClass,
+    IncidentKeyClass,
+    RoleKeyClass,
+    DomainKeyClass,
+    DataContractKeyClass,
+    AssertionKeyClass,
+    DatasetKeyClass,
     DataJobKeyClass,
+    DataHubSecretKeyClass,
+    OwnershipTypeKeyClass,
+    SchemaFieldKeyClass,
+    MLFeatureTableKeyClass,
+    DashboardKeyClass,
+    TelemetryKeyClass,
+    MonitorKeyClass,
+    MLFeatureKeyClass,
+    DataHubStepStateKeyClass,
+    ConstraintKeyClass,
+    DataHubViewKeyClass,
+    DataHubRetentionKeyClass,
+    MLPrimaryKeyKeyClass,
     DataFlowKeyClass,
+    CorpUserKeyClass,
+    InviteTokenKeyClass,
     DataPlatformKeyClass,
-    RecommendationModuleKeyClass,
-    TagKeyClass,
-    LinkPreviewKeyClass,
-    NotebookKeyClass,
-    DataHubAccessTokenKeyClass,
-    ERModelRelationshipKeyClass,
-    OwnershipTypeKeyClass,
-    QueryKeyClass,
-    AssertionKeyClass,
-    DataContractKeyClass,
-    DataHubPolicyKeyClass,
-    DataHubActionKeyClass,
-    GenericEntityKeyClass,
-    SubscriptionKeyClass,
+    AnomalyKeyClass,
     ContainerKeyClass,
-    DataProcessKeyClass,
-    DataHubViewKeyClass,
+    GenericEntityKeyClass,
     DataPlatformInstanceKeyClass,
-    RoleKeyClass,
-    FormKeyClass,
-    MLFeatureKeyClass,
-    PostKeyClass,
-    DatasetKeyClass,
-    MLModelKeyClass,
-    MLPrimaryKeyKeyClass,
-    MLModelDeploymentKeyClass,
-    MLFeatureTableKeyClass,
-    DataHubUpgradeKeyClass,
     GlossaryNodeKeyClass,
+    MLModelKeyClass,
+    TestKeyClass,
+    PostKeyClass,
+    GlossaryTermKeyClass,
+    ERModelRelationshipKeyClass,
+    ExecutionRequestKeyClass,
+    LinkPreviewKeyClass,
+    QueryKeyClass,
     DataHubPersonaKeyClass,
-    DataHubSecretKeyClass,
+    ChartKeyClass,
+    DataHubAccessTokenKeyClass,
+    DataProcessInstanceKeyClass,
+    CorpGroupKeyClass,
+    DataProcessKeyClass,
     DataHubIngestionSourceKeyClass,
-    ConstraintKeyClass,
+    SubscriptionKeyClass,
+    RecommendationModuleKeyClass,
     ActionRequestKeyClass,
-    DataHubRetentionKeyClass,
-    TestKeyClass,
-    ChartKeyClass,
-    MLModelGroupKeyClass,
-    TelemetryKeyClass,
-    DomainKeyClass,
-    MonitorKeyClass,
-    IncidentKeyClass,
-    GlossaryTermKeyClass,
-    AnomalyKeyClass,
-    InviteTokenKeyClass,
-    UsageFeaturesClass,
+    GlobalSettingsKeyClass,
+    StorageFeaturesClass,
     LineageFeaturesClass,
     CostFeaturesClass,
-    StorageFeaturesClass,
-    RecommendationModuleClass,
-    CorpUserStatusClass,
-    InviteTokenClass,
-    CorpUserSettingsClass,
-    CorpUserEditableInfoClass,
-    RoleMembershipClass,
-    CorpGroupInfoClass,
-    CorpUserInfoClass,
-    CorpGroupSettingsClass,
-    CorpGroupEditableInfoClass,
-    NativeGroupMembershipClass,
-    CorpUserCredentialsClass,
-    GroupMembershipClass,
-    DomainsClass,
-    DomainPropertiesClass,
-    PlatformResourceInfoClass,
-    PlatformResourceKeyClass,
+    UsageFeaturesClass,
     EntityTypeInfoClass,
     EntityTypeKeyClass,
-    SchemaFieldProfileClass,
-    DatasetPropertiesClass,
-    DatasetProfileClass,
-    PartitionsSummaryClass,
-    DatasetUsageStatisticsClass,
-    UpstreamLineageClass,
-    DatasetUpstreamLineageClass,
-    DatasetDeprecationClass,
-    ViewPropertiesClass,
-    EditableDatasetPropertiesClass,
-    SubscriptionInfoClass,
-    BatchTestRunEventClass,
-    TestInfoClass,
-    TestResultsClass,
-    ContainerClass,
-    EditableContainerPropertiesClass,
-    ContainerPropertiesClass,
-    ConstraintInfoClass,
-    DataProductPropertiesClass,
-    DataProductKeyClass,
-    BusinessAttributesClass,
-    BusinessAttributeInfoClass,
-    BusinessAttributeKeyClass,
-    NotebookContentClass,
-    EditableNotebookPropertiesClass,
-    NotebookInfoClass,
-    DataHubViewInfoClass,
-    VersionInfoClass,
-    EditableDataJobPropertiesClass,
-    DataFlowInfoClass,
-    EditableDataFlowPropertiesClass,
-    DataJobInfoClass,
-    DataJobInputOutputClass,
-    DatahubIngestionCheckpointClass,
-    DatahubIngestionRunSummaryClass
+    OwnershipTypeInfoClass,
+    DataPlatformInstancePropertiesClass,
+    DataHubAccessTokenInfoClass,
+    AiInferenceMetadataClass,
+    EntityInferenceMetadataClass
 ]
 
 ASPECT_NAME_MAP: Dict[str, Type[_Aspect]] = {
@@ -35917,325 +36122,326 @@ ASPECT_NAME_MAP: Dict[str, Type[_Aspect]] = {
     for aspect in ASPECT_CLASSES
 }
 
-from typing_extensions import TypedDict
-
-class AspectBag(TypedDict, total=False):
-    tagProperties: TagPropertiesClass
-    status: StatusClass
-    incidentsSummary: IncidentsSummaryClass
-    cost: CostClass
-    browsePaths: BrowsePathsClass
-    assertionsSummary: AssertionsSummaryClass
-    anomaliesSummary: AnomaliesSummaryClass
-    institutionalMemory: InstitutionalMemoryClass
-    proposals: ProposalsClass
-    subTypes: SubTypesClass
-    forms: FormsClass
+from typing_extensions import TypedDict
+
+class AspectBag(TypedDict, total=False):
+    dataHubRetentionConfig: DataHubRetentionConfigClass
+    incidentInfo: IncidentInfoClass
+    incidentActivityEvent: IncidentActivityEventClass
+    incidentNotificationDetails: IncidentNotificationDetailsClass
+    incidentSource: IncidentSourceClass
+    dataHubConnectionDetails: DataHubConnectionDetailsClass
     deprecation: DeprecationClass
-    globalTags: GlobalTagsClass
-    embed: EmbedClass
-    operation: OperationClass
-    dataPlatformInstance: DataPlatformInstanceClass
+    share: ShareClass
+    browsePathsV2: BrowsePathsV2Class
+    origin: OriginClass
     glossaryTerms: GlossaryTermsClass
+    cost: CostClass
+    access: AccessClass
+    subTypes: SubTypesClass
     siblings: SiblingsClass
+    incidentsSummary: IncidentsSummaryClass
+    institutionalMemory: InstitutionalMemoryClass
+    assertionsSummary: AssertionsSummaryClass
+    browsePaths: BrowsePathsClass
+    status: StatusClass
+    documentation: DocumentationClass
+    dataPlatformInstance: DataPlatformInstanceClass
+    anomaliesSummary: AnomaliesSummaryClass
     inputFields: InputFieldsClass
     displayProperties: DisplayPropertiesClass
+    proposals: ProposalsClass
     ownership: OwnershipClass
-    documentation: DocumentationClass
-    browsePathsV2: BrowsePathsV2Class
-    origin: OriginClass
-    share: ShareClass
-    access: AccessClass
-    dataHubAccessTokenInfo: DataHubAccessTokenInfoClass
+    operation: OperationClass
+    forms: FormsClass
+    embed: EmbedClass
+    globalTags: GlobalTagsClass
     globalSettingsInfo: GlobalSettingsInfoClass
-    aiInferenceMetadata: AiInferenceMetadataClass
-    entityInferenceMetadata: EntityInferenceMetadataClass
-    schemaMetadata: SchemaMetadataClass
-    schemaProposals: SchemaProposalsClass
-    editableSchemaMetadata: EditableSchemaMetadataClass
-    inferredNeighbors: InferredNeighborsClass
-    inferredMetadata: InferredMetadataClass
-    schemaFieldsInferredNeighbors: SchemaFieldsInferredNeighborsClass
-    schemaFieldsInferredMetadata: SchemaFieldsInferredMetadataClass
-    dataPlatformInstanceProperties: DataPlatformInstancePropertiesClass
-    erModelRelationshipProperties: ERModelRelationshipPropertiesClass
-    editableERModelRelationshipProperties: EditableERModelRelationshipPropertiesClass
+    constraintInfo: ConstraintInfoClass
+    structuredProperties: StructuredPropertiesClass
+    structuredPropertyKey: StructuredPropertyKeyClass
     propertyDefinition: StructuredPropertyDefinitionClass
     structuredPropertySettings: StructuredPropertySettingsClass
-    structuredPropertyKey: StructuredPropertyKeyClass
-    structuredProperties: StructuredPropertiesClass
-    roleProperties: RolePropertiesClass
-    actors: ActorsClass
-    dashboardUsageStatistics: DashboardUsageStatisticsClass
+    schemafieldInfo: SchemaFieldInfoClass
+    schemaFieldAliases: SchemaFieldAliasesClass
+    schemaFieldsInferredMetadata: SchemaFieldsInferredMetadataClass
+    schemaFieldsInferredNeighbors: SchemaFieldsInferredNeighborsClass
+    inferredMetadata: InferredMetadataClass
+    inferredNeighbors: InferredNeighborsClass
+    containerProperties: ContainerPropertiesClass
+    editableContainerProperties: EditableContainerPropertiesClass
+    container: ContainerClass
+    dataHubUpgradeResult: DataHubUpgradeResultClass
+    dataHubUpgradeRequest: DataHubUpgradeRequestClass
+    anomalyInfo: AnomalyInfoClass
+    businessAttributes: BusinessAttributesClass
+    businessAttributeInfo: BusinessAttributeInfoClass
+    businessAttributeKey: BusinessAttributeKeyClass
+    dataHubIngestionSourceInfo: DataHubIngestionSourceInfoClass
+    subscriptionInfo: SubscriptionInfoClass
+    tagProperties: TagPropertiesClass
     editableDashboardProperties: EditableDashboardPropertiesClass
+    dashboardUsageStatistics: DashboardUsageStatisticsClass
     dashboardInfo: DashboardInfoClass
-    ownershipTypeInfo: OwnershipTypeInfoClass
-    monitorTimeseriesState: MonitorTimeseriesStateClass
+    dataHubViewInfo: DataHubViewInfoClass
+    dataContractStatus: DataContractStatusClass
+    dataContractProperties: DataContractPropertiesClass
+    roleProperties: RolePropertiesClass
+    actors: ActorsClass
+    schemaMetadata: SchemaMetadataClass
+    schemaProposals: SchemaProposalsClass
+    editableSchemaMetadata: EditableSchemaMetadataClass
+    dataPlatformInfo: DataPlatformInfoClass
+    slackUserInfo: SlackUserInfoClass
     monitorInfo: MonitorInfoClass
+    monitorTimeseriesState: MonitorTimeseriesStateClass
+    dataHubPersonaInfo: DataHubPersonaInfoClass
+    dataHubPolicyInfo: DataHubPolicyInfoClass
+    dataHubRoleInfo: DataHubRoleInfoClass
+    dataFlowInfo: DataFlowInfoClass
+    dataJobInfo: DataJobInfoClass
+    versionInfo: VersionInfoClass
+    editableDataJobProperties: EditableDataJobPropertiesClass
+    editableDataFlowProperties: EditableDataFlowPropertiesClass
+    dataJobInputOutput: DataJobInputOutputClass
+    datahubIngestionRunSummary: DatahubIngestionRunSummaryClass
+    datahubIngestionCheckpoint: DatahubIngestionCheckpointClass
+    assertionSummary: AssertionSummaryClass
+    assertionInfo: AssertionInfoClass
+    assertionAnalyticsRunEvent: AssertionAnalyticsRunEventClass
+    assertionDryRunEvent: AssertionDryRunEventClass
+    assertionInferenceDetails: AssertionInferenceDetailsClass
+    assertionRunEvent: AssertionRunEventClass
+    assertionActions: AssertionActionsClass
     glossaryRelatedTerms: GlossaryRelatedTermsClass
     glossaryNodeInfo: GlossaryNodeInfoClass
     glossaryTermInfo: GlossaryTermInfoClass
-    dataHubPersonaInfo: DataHubPersonaInfoClass
-    schemaFieldAliases: SchemaFieldAliasesClass
-    schemafieldInfo: SchemaFieldInfoClass
+    domains: DomainsClass
+    domainProperties: DomainPropertiesClass
     actionRequestInfo: ActionRequestInfoClass
-    actionRequestArchived: ActionRequestArchivedClass
     actionRequestStatus: ActionRequestStatusClass
-    dataHubIngestionSourceInfo: DataHubIngestionSourceInfoClass
-    dataHubRetentionConfig: DataHubRetentionConfigClass
-    dataHubActionInfo: DataHubActionInfoClass
-    dataHubActionStatus: DataHubActionStatusClass
-    dataPlatformInfo: DataPlatformInfoClass
-    slackUserInfo: SlackUserInfoClass
-    postInfo: PostInfoClass
-    editableChartProperties: EditableChartPropertiesClass
-    chartInfo: ChartInfoClass
-    chartUsageStatistics: ChartUsageStatisticsClass
-    chartQuery: ChartQueryClass
-    formInfo: FormInfoClass
-    dynamicFormAssignment: DynamicFormAssignmentClass
-    queryUsageFeatures: QueryUsageFeaturesClass
-    queryProperties: QueryPropertiesClass
-    querySubjects: QuerySubjectsClass
-    queryUsageStatistics: QueryUsageStatisticsClass
-    dataHubExecutionRequestResult: ExecutionRequestResultClass
-    dataHubExecutionRequestSignal: ExecutionRequestSignalClass
-    dataHubExecutionRequestInput: ExecutionRequestInputClass
-    dataTypeKey: DataTypeKeyClass
-    dataTypeInfo: DataTypeInfoClass
-    dataProcessInstanceInput: DataProcessInstanceInputClass
-    dataProcessInstanceRunEvent: DataProcessInstanceRunEventClass
-    dataProcessInstanceOutput: DataProcessInstanceOutputClass
-    dataProcessInstanceProperties: DataProcessInstancePropertiesClass
-    dataProcessInstanceRelationships: DataProcessInstanceRelationshipsClass
-    dataProcessInfo: DataProcessInfoClass
-    dataHubRoleInfo: DataHubRoleInfoClass
-    dataHubPolicyInfo: DataHubPolicyInfoClass
+    actionRequestArchived: ActionRequestArchivedClass
+    dataHubStepStateProperties: DataHubStepStatePropertiesClass
+    corpUserSettings: CorpUserSettingsClass
+    corpUserEditableInfo: CorpUserEditableInfoClass
+    nativeGroupMembership: NativeGroupMembershipClass
+    corpUserStatus: CorpUserStatusClass
+    corpGroupInfo: CorpGroupInfoClass
+    roleMembership: RoleMembershipClass
+    inviteToken: InviteTokenClass
+    corpGroupSettings: CorpGroupSettingsClass
+    groupMembership: GroupMembershipClass
+    corpUserInfo: CorpUserInfoClass
+    corpUserCredentials: CorpUserCredentialsClass
+    corpGroupEditableInfo: CorpGroupEditableInfoClass
+    telemetryClientId: TelemetryClientIdClass
+    dataProductKey: DataProductKeyClass
+    dataProductProperties: DataProductPropertiesClass
+    notebookInfo: NotebookInfoClass
+    notebookContent: NotebookContentClass
+    editableNotebookProperties: EditableNotebookPropertiesClass
+    editableMlFeatureTableProperties: EditableMLFeatureTablePropertiesClass
+    mlHyperParam: MLHyperParamClass
+    mlFeatureProperties: MLFeaturePropertiesClass
+    mlPrimaryKeyProperties: MLPrimaryKeyPropertiesClass
+    mlModelProperties: MLModelPropertiesClass
+    mlTrainingRunProperties: MLTrainingRunPropertiesClass
+    editableMlModelGroupProperties: EditableMLModelGroupPropertiesClass
     mlModelDeploymentProperties: MLModelDeploymentPropertiesClass
-    mlFeatureTableProperties: MLFeatureTablePropertiesClass
-    mlModelTrainingData: TrainingDataClass
-    intendedUse: IntendedUseClass
     editableMlFeatureProperties: EditableMLFeaturePropertiesClass
-    mlModelEvaluationData: EvaluationDataClass
-    mlMetric: MLMetricClass
-    mlFeatureProperties: MLFeaturePropertiesClass
-    editableMlFeatureTableProperties: EditableMLFeatureTablePropertiesClass
-    mlModelMetrics: MetricsClass
-    editableMlPrimaryKeyProperties: EditableMLPrimaryKeyPropertiesClass
-    sourceCode: SourceCodeClass
     mlModelEthicalConsiderations: EthicalConsiderationsClass
-    mlPrimaryKeyProperties: MLPrimaryKeyPropertiesClass
+    mlModelQuantitativeAnalyses: QuantitativeAnalysesClass
+    sourceCode: SourceCodeClass
+    editableMlPrimaryKeyProperties: EditableMLPrimaryKeyPropertiesClass
     mlModelFactorPrompts: MLModelFactorPromptsClass
-    editableMlModelGroupProperties: EditableMLModelGroupPropertiesClass
-    mlModelProperties: MLModelPropertiesClass
+    mlMetric: MLMetricClass
+    mlModelMetrics: MetricsClass
+    mlModelTrainingData: TrainingDataClass
+    intendedUse: IntendedUseClass
+    mlModelEvaluationData: EvaluationDataClass
+    mlFeatureTableProperties: MLFeatureTablePropertiesClass
     mlModelCaveatsAndRecommendations: CaveatsAndRecommendationsClass
-    mlHyperParam: MLHyperParamClass
-    mlModelQuantitativeAnalyses: QuantitativeAnalysesClass
-    mlModelGroupProperties: MLModelGroupPropertiesClass
     editableMlModelProperties: EditableMLModelPropertiesClass
-    incidentNotificationDetails: IncidentNotificationDetailsClass
-    incidentActivityEvent: IncidentActivityEventClass
-    incidentSource: IncidentSourceClass
-    incidentInfo: IncidentInfoClass
+    mlModelGroupProperties: MLModelGroupPropertiesClass
+    queryProperties: QueryPropertiesClass
+    queryUsageFeatures: QueryUsageFeaturesClass
+    querySubjects: QuerySubjectsClass
+    queryUsageStatistics: QueryUsageStatisticsClass
     linkPreviewInfo: LinkPreviewInfoClass
-    anomalyInfo: AnomalyInfoClass
-    dataHubConnectionDetails: DataHubConnectionDetailsClass
-    dataContractProperties: DataContractPropertiesClass
-    dataContractStatus: DataContractStatusClass
-    dataHubStepStateProperties: DataHubStepStatePropertiesClass
-    assertionDryRunEvent: AssertionDryRunEventClass
-    assertionRunEvent: AssertionRunEventClass
-    assertionSummary: AssertionSummaryClass
-    assertionInfo: AssertionInfoClass
-    assertionInferenceDetails: AssertionInferenceDetailsClass
-    assertionAnalyticsRunEvent: AssertionAnalyticsRunEventClass
-    assertionActions: AssertionActionsClass
-    telemetryClientId: TelemetryClientIdClass
     dataHubSecretValue: DataHubSecretValueClass
-    dataHubUpgradeRequest: DataHubUpgradeRequestClass
-    dataHubUpgradeResult: DataHubUpgradeResultClass
-    corpGroupKey: CorpGroupKeyClass
-    globalSettingsKey: GlobalSettingsKeyClass
-    schemaFieldKey: SchemaFieldKeyClass
-    dataHubStepStateKey: DataHubStepStateKeyClass
-    corpUserKey: CorpUserKeyClass
-    dataHubConnectionKey: DataHubConnectionKeyClass
-    dashboardKey: DashboardKeyClass
+    datasetUpstreamLineage: DatasetUpstreamLineageClass
+    upstreamLineage: UpstreamLineageClass
+    datasetProfile: DatasetProfileClass
+    schemaFieldProfile: SchemaFieldProfileClass
+    partitionsSummary: PartitionsSummaryClass
+    datasetProperties: DatasetPropertiesClass
+    datasetDeprecation: DatasetDeprecationClass
+    editableDatasetProperties: EditableDatasetPropertiesClass
+    datasetUsageStatistics: DatasetUsageStatisticsClass
+    viewProperties: ViewPropertiesClass
+    formInfo: FormInfoClass
+    dynamicFormAssignment: DynamicFormAssignmentClass
+    dataTypeInfo: DataTypeInfoClass
+    dataTypeKey: DataTypeKeyClass
+    testResults: TestResultsClass
+    batchTestRunEvent: BatchTestRunEventClass
+    testInfo: TestInfoClass
+    platformResourceInfo: PlatformResourceInfoClass
+    platformResourceKey: PlatformResourceKeyClass
+    postInfo: PostInfoClass
+    dataProcessInstanceProperties: DataProcessInstancePropertiesClass
+    dataProcessInfo: DataProcessInfoClass
+    dataProcessInstanceRelationships: DataProcessInstanceRelationshipsClass
+    dataProcessInstanceRunEvent: DataProcessInstanceRunEventClass
+    dataProcessInstanceInput: DataProcessInstanceInputClass
+    dataProcessInstanceOutput: DataProcessInstanceOutputClass
+    editableChartProperties: EditableChartPropertiesClass
+    chartInfo: ChartInfoClass
+    chartQuery: ChartQueryClass
+    chartUsageStatistics: ChartUsageStatisticsClass
+    editableERModelRelationshipProperties: EditableERModelRelationshipPropertiesClass
+    erModelRelationshipProperties: ERModelRelationshipPropertiesClass
+    dataHubActionInfo: DataHubActionInfoClass
+    dataHubActionStatus: DataHubActionStatusClass
+    dataHubExecutionRequestResult: ExecutionRequestResultClass
+    dataHubExecutionRequestSignal: ExecutionRequestSignalClass
+    dataHubExecutionRequestInput: ExecutionRequestInputClass
+    recommendationModule: RecommendationModuleClass
+    mlModelGroupKey: MLModelGroupKeyClass
+    dataHubPolicyKey: DataHubPolicyKeyClass
+    tagKey: TagKeyClass
+    mlModelDeploymentKey: MLModelDeploymentKeyClass
+    dataHubActionKey: DataHubActionKeyClass
     dataHubRoleKey: DataHubRoleKeyClass
-    dataHubExecutionRequestKey: ExecutionRequestKeyClass
-    dataProcessInstanceKey: DataProcessInstanceKeyClass
+    dataHubUpgradeKey: DataHubUpgradeKeyClass
+    formKey: FormKeyClass
+    notebookKey: NotebookKeyClass
+    dataHubConnectionKey: DataHubConnectionKeyClass
+    incidentKey: IncidentKeyClass
+    roleKey: RoleKeyClass
+    domainKey: DomainKeyClass
+    dataContractKey: DataContractKeyClass
+    assertionKey: AssertionKeyClass
+    datasetKey: DatasetKeyClass
     dataJobKey: DataJobKeyClass
+    dataHubSecretKey: DataHubSecretKeyClass
+    ownershipTypeKey: OwnershipTypeKeyClass
+    schemaFieldKey: SchemaFieldKeyClass
+    mlFeatureTableKey: MLFeatureTableKeyClass
+    dashboardKey: DashboardKeyClass
+    telemetryKey: TelemetryKeyClass
+    monitorKey: MonitorKeyClass
+    mlFeatureKey: MLFeatureKeyClass
+    dataHubStepStateKey: DataHubStepStateKeyClass
+    constraintKey: ConstraintKeyClass
+    dataHubViewKey: DataHubViewKeyClass
+    dataHubRetentionKey: DataHubRetentionKeyClass
+    mlPrimaryKeyKey: MLPrimaryKeyKeyClass
     dataFlowKey: DataFlowKeyClass
+    corpUserKey: CorpUserKeyClass
+    inviteTokenKey: InviteTokenKeyClass
     dataPlatformKey: DataPlatformKeyClass
-    recommendationModuleKey: RecommendationModuleKeyClass
-    tagKey: TagKeyClass
-    linkPreviewKey: LinkPreviewKeyClass
-    notebookKey: NotebookKeyClass
-    dataHubAccessTokenKey: DataHubAccessTokenKeyClass
-    erModelRelationshipKey: ERModelRelationshipKeyClass
-    ownershipTypeKey: OwnershipTypeKeyClass
-    queryKey: QueryKeyClass
-    assertionKey: AssertionKeyClass
-    dataContractKey: DataContractKeyClass
-    dataHubPolicyKey: DataHubPolicyKeyClass
-    dataHubActionKey: DataHubActionKeyClass
-    genericEntityKey: GenericEntityKeyClass
-    subscriptionKey: SubscriptionKeyClass
+    anomalyKey: AnomalyKeyClass
     containerKey: ContainerKeyClass
-    dataProcessKey: DataProcessKeyClass
-    dataHubViewKey: DataHubViewKeyClass
+    genericEntityKey: GenericEntityKeyClass
     dataPlatformInstanceKey: DataPlatformInstanceKeyClass
-    roleKey: RoleKeyClass
-    formKey: FormKeyClass
-    mlFeatureKey: MLFeatureKeyClass
-    postKey: PostKeyClass
-    datasetKey: DatasetKeyClass
-    mlModelKey: MLModelKeyClass
-    mlPrimaryKeyKey: MLPrimaryKeyKeyClass
-    mlModelDeploymentKey: MLModelDeploymentKeyClass
-    mlFeatureTableKey: MLFeatureTableKeyClass
-    dataHubUpgradeKey: DataHubUpgradeKeyClass
     glossaryNodeKey: GlossaryNodeKeyClass
+    mlModelKey: MLModelKeyClass
+    testKey: TestKeyClass
+    postKey: PostKeyClass
+    glossaryTermKey: GlossaryTermKeyClass
+    erModelRelationshipKey: ERModelRelationshipKeyClass
+    dataHubExecutionRequestKey: ExecutionRequestKeyClass
+    linkPreviewKey: LinkPreviewKeyClass
+    queryKey: QueryKeyClass
     dataHubPersonaKey: DataHubPersonaKeyClass
-    dataHubSecretKey: DataHubSecretKeyClass
+    chartKey: ChartKeyClass
+    dataHubAccessTokenKey: DataHubAccessTokenKeyClass
+    dataProcessInstanceKey: DataProcessInstanceKeyClass
+    corpGroupKey: CorpGroupKeyClass
+    dataProcessKey: DataProcessKeyClass
     dataHubIngestionSourceKey: DataHubIngestionSourceKeyClass
-    constraintKey: ConstraintKeyClass
+    subscriptionKey: SubscriptionKeyClass
+    recommendationModuleKey: RecommendationModuleKeyClass
     actionRequestKey: ActionRequestKeyClass
-    dataHubRetentionKey: DataHubRetentionKeyClass
-    testKey: TestKeyClass
-    chartKey: ChartKeyClass
-    mlModelGroupKey: MLModelGroupKeyClass
-    telemetryKey: TelemetryKeyClass
-    domainKey: DomainKeyClass
-    monitorKey: MonitorKeyClass
-    incidentKey: IncidentKeyClass
-    glossaryTermKey: GlossaryTermKeyClass
-    anomalyKey: AnomalyKeyClass
-    inviteTokenKey: InviteTokenKeyClass
-    usageFeatures: UsageFeaturesClass
+    globalSettingsKey: GlobalSettingsKeyClass
+    storageFeatures: StorageFeaturesClass
     lineageFeatures: LineageFeaturesClass
     costFeatures: CostFeaturesClass
-    storageFeatures: StorageFeaturesClass
-    recommendationModule: RecommendationModuleClass
-    corpUserStatus: CorpUserStatusClass
-    inviteToken: InviteTokenClass
-    corpUserSettings: CorpUserSettingsClass
-    corpUserEditableInfo: CorpUserEditableInfoClass
-    roleMembership: RoleMembershipClass
-    corpGroupInfo: CorpGroupInfoClass
-    corpUserInfo: CorpUserInfoClass
-    corpGroupSettings: CorpGroupSettingsClass
-    corpGroupEditableInfo: CorpGroupEditableInfoClass
-    nativeGroupMembership: NativeGroupMembershipClass
-    corpUserCredentials: CorpUserCredentialsClass
-    groupMembership: GroupMembershipClass
-    domains: DomainsClass
-    domainProperties: DomainPropertiesClass
-    platformResourceInfo: PlatformResourceInfoClass
-    platformResourceKey: PlatformResourceKeyClass
+    usageFeatures: UsageFeaturesClass
     entityTypeInfo: EntityTypeInfoClass
     entityTypeKey: EntityTypeKeyClass
-    schemaFieldProfile: SchemaFieldProfileClass
-    datasetProperties: DatasetPropertiesClass
-    datasetProfile: DatasetProfileClass
-    partitionsSummary: PartitionsSummaryClass
-    datasetUsageStatistics: DatasetUsageStatisticsClass
-    upstreamLineage: UpstreamLineageClass
-    datasetUpstreamLineage: DatasetUpstreamLineageClass
-    datasetDeprecation: DatasetDeprecationClass
-    viewProperties: ViewPropertiesClass
-    editableDatasetProperties: EditableDatasetPropertiesClass
-    subscriptionInfo: SubscriptionInfoClass
-    batchTestRunEvent: BatchTestRunEventClass
-    testInfo: TestInfoClass
-    testResults: TestResultsClass
-    container: ContainerClass
-    editableContainerProperties: EditableContainerPropertiesClass
-    containerProperties: ContainerPropertiesClass
-    constraintInfo: ConstraintInfoClass
-    dataProductProperties: DataProductPropertiesClass
-    dataProductKey: DataProductKeyClass
-    businessAttributes: BusinessAttributesClass
-    businessAttributeInfo: BusinessAttributeInfoClass
-    businessAttributeKey: BusinessAttributeKeyClass
-    notebookContent: NotebookContentClass
-    editableNotebookProperties: EditableNotebookPropertiesClass
-    notebookInfo: NotebookInfoClass
-    dataHubViewInfo: DataHubViewInfoClass
-    versionInfo: VersionInfoClass
-    editableDataJobProperties: EditableDataJobPropertiesClass
-    dataFlowInfo: DataFlowInfoClass
-    editableDataFlowProperties: EditableDataFlowPropertiesClass
-    dataJobInfo: DataJobInfoClass
-    dataJobInputOutput: DataJobInputOutputClass
-    datahubIngestionCheckpoint: DatahubIngestionCheckpointClass
-    datahubIngestionRunSummary: DatahubIngestionRunSummaryClass
+    ownershipTypeInfo: OwnershipTypeInfoClass
+    dataPlatformInstanceProperties: DataPlatformInstancePropertiesClass
+    dataHubAccessTokenInfo: DataHubAccessTokenInfoClass
+    aiInferenceMetadata: AiInferenceMetadataClass
+    entityInferenceMetadata: EntityInferenceMetadataClass
 
 
 KEY_ASPECTS: Dict[str, Type[_Aspect]] = {
     'structuredProperty': StructuredPropertyKeyClass,
+    'businessAttribute': BusinessAttributeKeyClass,
+    'dataProduct': DataProductKeyClass,
     'dataType': DataTypeKeyClass,
-    'corpGroup': CorpGroupKeyClass,
-    'globalSettings': GlobalSettingsKeyClass,
-    'schemaField': SchemaFieldKeyClass,
-    'dataHubStepState': DataHubStepStateKeyClass,
-    'corpuser': CorpUserKeyClass,
-    'dataHubConnection': DataHubConnectionKeyClass,
-    'dashboard': DashboardKeyClass,
+    'platformResource': PlatformResourceKeyClass,
+    'mlModelGroup': MLModelGroupKeyClass,
+    'dataHubPolicy': DataHubPolicyKeyClass,
+    'tag': TagKeyClass,
+    'mlModelDeployment': MLModelDeploymentKeyClass,
+    'dataHubAction': DataHubActionKeyClass,
     'dataHubRole': DataHubRoleKeyClass,
-    'dataHubExecutionRequest': ExecutionRequestKeyClass,
-    'dataProcessInstance': DataProcessInstanceKeyClass,
+    'dataHubUpgrade': DataHubUpgradeKeyClass,
+    'form': FormKeyClass,
+    'notebook': NotebookKeyClass,
+    'dataHubConnection': DataHubConnectionKeyClass,
+    'incident': IncidentKeyClass,
+    'role': RoleKeyClass,
+    'domain': DomainKeyClass,
+    'dataContract': DataContractKeyClass,
+    'assertion': AssertionKeyClass,
+    'dataset': DatasetKeyClass,
     'dataJob': DataJobKeyClass,
+    'dataHubSecret': DataHubSecretKeyClass,
+    'ownershipType': OwnershipTypeKeyClass,
+    'schemaField': SchemaFieldKeyClass,
+    'mlFeatureTable': MLFeatureTableKeyClass,
+    'dashboard': DashboardKeyClass,
+    'telemetry': TelemetryKeyClass,
+    'monitor': MonitorKeyClass,
+    'mlFeature': MLFeatureKeyClass,
+    'dataHubStepState': DataHubStepStateKeyClass,
+    'constraint': ConstraintKeyClass,
+    'dataHubView': DataHubViewKeyClass,
+    'dataHubRetention': DataHubRetentionKeyClass,
+    'mlPrimaryKey': MLPrimaryKeyKeyClass,
     'dataFlow': DataFlowKeyClass,
+    'corpuser': CorpUserKeyClass,
+    'inviteToken': InviteTokenKeyClass,
     'dataPlatform': DataPlatformKeyClass,
-    'recommendationModule': RecommendationModuleKeyClass,
-    'tag': TagKeyClass,
-    'linkPreview': LinkPreviewKeyClass,
-    'notebook': NotebookKeyClass,
-    'dataHubAccessToken': DataHubAccessTokenKeyClass,
-    'erModelRelationship': ERModelRelationshipKeyClass,
-    'ownershipType': OwnershipTypeKeyClass,
-    'query': QueryKeyClass,
-    'assertion': AssertionKeyClass,
-    'dataContract': DataContractKeyClass,
-    'dataHubPolicy': DataHubPolicyKeyClass,
-    'dataHubAction': DataHubActionKeyClass,
-    'subscription': SubscriptionKeyClass,
+    'anomaly': AnomalyKeyClass,
     'container': ContainerKeyClass,
-    'dataProcess': DataProcessKeyClass,
-    'dataHubView': DataHubViewKeyClass,
     'dataPlatformInstance': DataPlatformInstanceKeyClass,
-    'role': RoleKeyClass,
-    'form': FormKeyClass,
-    'mlFeature': MLFeatureKeyClass,
-    'post': PostKeyClass,
-    'dataset': DatasetKeyClass,
-    'mlModel': MLModelKeyClass,
-    'mlPrimaryKey': MLPrimaryKeyKeyClass,
-    'mlModelDeployment': MLModelDeploymentKeyClass,
-    'mlFeatureTable': MLFeatureTableKeyClass,
-    'dataHubUpgrade': DataHubUpgradeKeyClass,
     'glossaryNode': GlossaryNodeKeyClass,
+    'mlModel': MLModelKeyClass,
+    'test': TestKeyClass,
+    'post': PostKeyClass,
+    'glossaryTerm': GlossaryTermKeyClass,
+    'erModelRelationship': ERModelRelationshipKeyClass,
+    'dataHubExecutionRequest': ExecutionRequestKeyClass,
+    'linkPreview': LinkPreviewKeyClass,
+    'query': QueryKeyClass,
     'dataHubPersona': DataHubPersonaKeyClass,
-    'dataHubSecret': DataHubSecretKeyClass,
+    'chart': ChartKeyClass,
+    'dataHubAccessToken': DataHubAccessTokenKeyClass,
+    'dataProcessInstance': DataProcessInstanceKeyClass,
+    'corpGroup': CorpGroupKeyClass,
+    'dataProcess': DataProcessKeyClass,
     'dataHubIngestionSource': DataHubIngestionSourceKeyClass,
-    'constraint': ConstraintKeyClass,
+    'subscription': SubscriptionKeyClass,
+    'recommendationModule': RecommendationModuleKeyClass,
     'actionRequest': ActionRequestKeyClass,
-    'dataHubRetention': DataHubRetentionKeyClass,
-    'test': TestKeyClass,
-    'chart': ChartKeyClass,
-    'mlModelGroup': MLModelGroupKeyClass,
-    'telemetry': TelemetryKeyClass,
-    'domain': DomainKeyClass,
-    'monitor': MonitorKeyClass,
-    'incident': IncidentKeyClass,
-    'glossaryTerm': GlossaryTermKeyClass,
-    'anomaly': AnomalyKeyClass,
-    'inviteToken': InviteTokenKeyClass,
-    'platformResource': PlatformResourceKeyClass,
-    'entityType': EntityTypeKeyClass,
-    'dataProduct': DataProductKeyClass,
-    'businessAttribute': BusinessAttributeKeyClass
+    'globalSettings': GlobalSettingsKeyClass,
+    'entityType': EntityTypeKeyClass
 }
 
 # fmt: on