acryl-datahub 0.15.0.2rc4__py3-none-any.whl → 0.15.0.2rc6__py3-none-any.whl

datahub/metadata/_schema_classes.py

@@ -6142,19 +6142,147 @@ class TimeStampClass(DictWrapper):
         self._inner_dict['actor'] = value
     
     
+class VersionPropertiesClass(_Aspect):
+    """Properties about a versioned asset i.e. dataset, ML Model, etc."""
+
+
+    ASPECT_NAME = 'versionProperties'
+    ASPECT_INFO = {}
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.common.VersionProperties")
+
+    def __init__(self,
+        versionSet: str,
+        version: "VersionTagClass",
+        sortId: str,
+        aliases: Optional[List["VersionTagClass"]]=None,
+        comment: Union[None, str]=None,
+        sourceCreatedTimestamp: Union[None, "AuditStampClass"]=None,
+        metadataCreatedTimestamp: Union[None, "AuditStampClass"]=None,
+        isLatest: Union[None, bool]=None,
+    ):
+        super().__init__()
+        
+        self.versionSet = versionSet
+        self.version = version
+        if aliases is None:
+            # default: []
+            self.aliases = list()
+        else:
+            self.aliases = aliases
+        self.comment = comment
+        self.sortId = sortId
+        self.sourceCreatedTimestamp = sourceCreatedTimestamp
+        self.metadataCreatedTimestamp = metadataCreatedTimestamp
+        self.isLatest = isLatest
+    
+    def _restore_defaults(self) -> None:
+        self.versionSet = str()
+        self.version = VersionTagClass._construct_with_defaults()
+        self.aliases = list()
+        self.comment = self.RECORD_SCHEMA.fields_dict["comment"].default
+        self.sortId = str()
+        self.sourceCreatedTimestamp = self.RECORD_SCHEMA.fields_dict["sourceCreatedTimestamp"].default
+        self.metadataCreatedTimestamp = self.RECORD_SCHEMA.fields_dict["metadataCreatedTimestamp"].default
+        self.isLatest = self.RECORD_SCHEMA.fields_dict["isLatest"].default
+    
+    
+    @property
+    def versionSet(self) -> str:
+        """The linked Version Set entity that ties multiple versioned assets together"""
+        return self._inner_dict.get('versionSet')  # type: ignore
+    
+    @versionSet.setter
+    def versionSet(self, value: str) -> None:
+        self._inner_dict['versionSet'] = value
+    
+    
+    @property
+    def version(self) -> "VersionTagClass":
+        """Label for this versioned asset, is unique within a version set"""
+        return self._inner_dict.get('version')  # type: ignore
+    
+    @version.setter
+    def version(self, value: "VersionTagClass") -> None:
+        self._inner_dict['version'] = value
+    
+    
+    @property
+    def aliases(self) -> List["VersionTagClass"]:
+        """Associated aliases for this versioned asset"""
+        return self._inner_dict.get('aliases')  # type: ignore
+    
+    @aliases.setter
+    def aliases(self, value: List["VersionTagClass"]) -> None:
+        self._inner_dict['aliases'] = value
+    
+    
+    @property
+    def comment(self) -> Union[None, str]:
+        """Comment documenting what this version was created for, changes, or represents"""
+        return self._inner_dict.get('comment')  # type: ignore
+    
+    @comment.setter
+    def comment(self, value: Union[None, str]) -> None:
+        self._inner_dict['comment'] = value
+    
+    
+    @property
+    def sortId(self) -> str:
+        """Sort identifier that determines where a version lives in the order of the Version Set.
+    What this looks like depends on the Version Scheme. For sort ids generated by DataHub we use an 8 character string representation."""
+        return self._inner_dict.get('sortId')  # type: ignore
+    
+    @sortId.setter
+    def sortId(self, value: str) -> None:
+        self._inner_dict['sortId'] = value
+    
+    
+    @property
+    def sourceCreatedTimestamp(self) -> Union[None, "AuditStampClass"]:
+        """Timestamp reflecting when this asset version was created in the source system."""
+        return self._inner_dict.get('sourceCreatedTimestamp')  # type: ignore
+    
+    @sourceCreatedTimestamp.setter
+    def sourceCreatedTimestamp(self, value: Union[None, "AuditStampClass"]) -> None:
+        self._inner_dict['sourceCreatedTimestamp'] = value
+    
+    
+    @property
+    def metadataCreatedTimestamp(self) -> Union[None, "AuditStampClass"]:
+        """Timestamp reflecting when the metadata for this version was created in DataHub"""
+        return self._inner_dict.get('metadataCreatedTimestamp')  # type: ignore
+    
+    @metadataCreatedTimestamp.setter
+    def metadataCreatedTimestamp(self, value: Union[None, "AuditStampClass"]) -> None:
+        self._inner_dict['metadataCreatedTimestamp'] = value
+    
+    
+    @property
+    def isLatest(self) -> Union[None, bool]:
+        """Marks whether this version is currently the latest. Set by a side effect and should not be modified by API."""
+        return self._inner_dict.get('isLatest')  # type: ignore
+    
+    @isLatest.setter
+    def isLatest(self, value: Union[None, bool]) -> None:
+        self._inner_dict['isLatest'] = value
+    
+    
 class VersionTagClass(DictWrapper):
     """A resource-defined string representing the resource state for the purpose of concurrency control"""
     
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.common.VersionTag")
     def __init__(self,
         versionTag: Union[None, str]=None,
+        metadataAttribution: Union[None, "MetadataAttributionClass"]=None,
     ):
         super().__init__()
         
         self.versionTag = versionTag
+        self.metadataAttribution = metadataAttribution
     
     def _restore_defaults(self) -> None:
         self.versionTag = self.RECORD_SCHEMA.fields_dict["versionTag"].default
+        self.metadataAttribution = self.RECORD_SCHEMA.fields_dict["metadataAttribution"].default
     
     
     @property
@@ -6167,6 +6295,16 @@ class VersionTagClass(DictWrapper):
         self._inner_dict['versionTag'] = value
     
     
+    @property
+    def metadataAttribution(self) -> Union[None, "MetadataAttributionClass"]:
+        # No docs available.
+        return self._inner_dict.get('metadataAttribution')  # type: ignore
+    
+    @metadataAttribution.setter
+    def metadataAttribution(self, value: Union[None, "MetadataAttributionClass"]) -> None:
+        self._inner_dict['metadataAttribution'] = value
+    
+    
 class WindowDurationClass(object):
     """Enum to define the length of a bucket when doing aggregations"""
     
@@ -14876,7 +15014,7 @@ class DatasetKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'datasetKey'
-    ASPECT_INFO = {'keyForEntity': 'dataset', 'entityCategory': 'core', 'entityAspects': ['viewProperties', 'subTypes', 'datasetProfile', 'datasetUsageStatistics', 'operation', 'domains', 'schemaMetadata', 'status', 'container', 'deprecation', 'testResults', 'siblings', 'embed', 'incidentsSummary', 'datasetProperties', 'editableDatasetProperties', 'datasetDeprecation', 'datasetUpstreamLineage', 'upstreamLineage', 'institutionalMemory', 'ownership', 'editableSchemaMetadata', 'globalTags', 'glossaryTerms', 'browsePaths', 'dataPlatformInstance', 'browsePathsV2', 'access', 'structuredProperties', 'forms', 'partitionsSummary'], 'entityDoc': 'Datasets represent logical or physical data assets stored or represented in various data platforms. Tables, Views, Streams are all instances of datasets.'}
+    ASPECT_INFO = {'keyForEntity': 'dataset', 'entityCategory': 'core', 'entityAspects': ['viewProperties', 'subTypes', 'datasetProfile', 'datasetUsageStatistics', 'operation', 'domains', 'schemaMetadata', 'status', 'container', 'deprecation', 'testResults', 'siblings', 'embed', 'incidentsSummary', 'datasetProperties', 'editableDatasetProperties', 'datasetDeprecation', 'datasetUpstreamLineage', 'upstreamLineage', 'institutionalMemory', 'ownership', 'editableSchemaMetadata', 'globalTags', 'glossaryTerms', 'browsePaths', 'dataPlatformInstance', 'browsePathsV2', 'access', 'structuredProperties', 'forms', 'partitionsSummary', 'versionProperties'], 'entityDoc': 'Datasets represent logical or physical data assets stored or represented in various data platforms. Tables, Views, Streams are all instances of datasets.'}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.DatasetKey")
 
     def __init__(self,
@@ -15386,7 +15524,7 @@ class MLModelKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'mlModelKey'
-    ASPECT_INFO = {'keyForEntity': 'mlModel', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlModelProperties', 'domains', 'ownership', 'mlModelProperties', 'intendedUse', 'mlModelFactorPrompts', 'mlModelMetrics', 'mlModelEvaluationData', 'mlModelTrainingData', 'mlModelQuantitativeAnalyses', 'mlModelEthicalConsiderations', 'mlModelCaveatsAndRecommendations', 'institutionalMemory', 'sourceCode', 'status', 'cost', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults']}
+    ASPECT_INFO = {'keyForEntity': 'mlModel', 'entityCategory': 'core', 'entityAspects': ['glossaryTerms', 'editableMlModelProperties', 'domains', 'ownership', 'mlModelProperties', 'intendedUse', 'mlModelFactorPrompts', 'mlModelMetrics', 'mlModelEvaluationData', 'mlModelTrainingData', 'mlModelQuantitativeAnalyses', 'mlModelEthicalConsiderations', 'mlModelCaveatsAndRecommendations', 'institutionalMemory', 'sourceCode', 'status', 'cost', 'deprecation', 'browsePaths', 'globalTags', 'dataPlatformInstance', 'browsePathsV2', 'structuredProperties', 'forms', 'testResults', 'versionProperties']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.MLModelKey")
 
     def __init__(self,
@@ -15766,6 +15904,48 @@ class TestKeyClass(_Aspect):
         self._inner_dict['id'] = value
     
     
+class VersionSetKeyClass(_Aspect):
+    """Key for a Version Set entity"""
+
+
+    ASPECT_NAME = 'versionSetKey'
+    ASPECT_INFO = {'keyForEntity': 'versionSet', 'entityCategory': 'core', 'entityAspects': ['versionSetProperties']}
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.VersionSetKey")
+
+    def __init__(self,
+        id: str,
+        entityType: str,
+    ):
+        super().__init__()
+        
+        self.id = id
+        self.entityType = entityType
+    
+    def _restore_defaults(self) -> None:
+        self.id = str()
+        self.entityType = str()
+    
+    
+    @property
+    def id(self) -> str:
+        """ID of the Version Set, generated from platform + asset id / name"""
+        return self._inner_dict.get('id')  # type: ignore
+    
+    @id.setter
+    def id(self, value: str) -> None:
+        self._inner_dict['id'] = value
+    
+    
+    @property
+    def entityType(self) -> str:
+        """Type of entities included in version set, limits to a single entity type between linked versioned entities"""
+        return self._inner_dict.get('entityType')  # type: ignore
+    
+    @entityType.setter
+    def entityType(self, value: str) -> None:
+        self._inner_dict['entityType'] = value
+    
+    
 class ConditionClass(object):
     """The matching condition in a filter criterion"""
     
@@ -17769,12 +17949,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,
-        trainingJobs: Union[None, List[str]]=None,
         version: Union[None, "VersionTagClass"]=None,
     ):
         super().__init__()
@@ -17784,22 +17965,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.trainingJobs = trainingJobs
         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.trainingJobs = self.RECORD_SCHEMA.fields_dict["trainingJobs"].default
         self.version = self.RECORD_SCHEMA.fields_dict["version"].default
     
     
@@ -17813,6 +17996,26 @@ 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"""
@@ -17863,16 +18066,6 @@ class MLModelGroupPropertiesClass(_Aspect):
         self._inner_dict['lastModified'] = value
     
     
-    @property
-    def trainingJobs(self) -> Union[None, List[str]]:
-        """List of jobs (if any) used to train the model group. Visible in Lineage."""
-        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 version(self) -> Union[None, "VersionTagClass"]:
         """Version of the MLModelGroup"""
@@ -17894,6 +18087,8 @@ 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,
@@ -17908,8 +18103,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__()
@@ -17920,6 +18113,8 @@ 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
@@ -17938,13 +18133,13 @@ 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
@@ -17959,8 +18154,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
     
     
@@ -17984,6 +18177,26 @@ 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"""
@@ -18126,26 +18339,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. 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 (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"""
@@ -24549,6 +24742,71 @@ class UserUsageCountsClass(DictWrapper):
         self._inner_dict['userEmail'] = value
     
     
+class VersionSetPropertiesClass(_Aspect):
+    # No docs available.
+
+
+    ASPECT_NAME = 'versionSetProperties'
+    ASPECT_INFO = {}
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.versionset.VersionSetProperties")
+
+    def __init__(self,
+        latest: str,
+        versioningScheme: Union[str, "VersioningSchemeClass"],
+        customProperties: Optional[Dict[str, str]]=None,
+    ):
+        super().__init__()
+        
+        if customProperties is None:
+            # default: {}
+            self.customProperties = dict()
+        else:
+            self.customProperties = customProperties
+        self.latest = latest
+        self.versioningScheme = versioningScheme
+    
+    def _restore_defaults(self) -> None:
+        self.customProperties = dict()
+        self.latest = str()
+        self.versioningScheme = VersioningSchemeClass.ALPHANUMERIC_GENERATED_BY_DATAHUB
+    
+    
+    @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 latest(self) -> str:
+        """The latest versioned entity linked to in this version set"""
+        return self._inner_dict.get('latest')  # type: ignore
+    
+    @latest.setter
+    def latest(self, value: str) -> None:
+        self._inner_dict['latest'] = value
+    
+    
+    @property
+    def versioningScheme(self) -> Union[str, "VersioningSchemeClass"]:
+        """What versioning scheme is being utilized for the versioned entities sort criterion. Static once set"""
+        return self._inner_dict.get('versioningScheme')  # type: ignore
+    
+    @versioningScheme.setter
+    def versioningScheme(self, value: Union[str, "VersioningSchemeClass"]) -> None:
+        self._inner_dict['versioningScheme'] = value
+    
+    
+class VersioningSchemeClass(object):
+    # No docs available.
+    
+    ALPHANUMERIC_GENERATED_BY_DATAHUB = "ALPHANUMERIC_GENERATED_BY_DATAHUB"
+    
+    
 class DataHubViewDefinitionClass(DictWrapper):
     """A View definition."""
     
@@ -24815,6 +25073,7 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.common.SubTypes': SubTypesClass,
     'com.linkedin.pegasus2avro.common.TagAssociation': TagAssociationClass,
     'com.linkedin.pegasus2avro.common.TimeStamp': TimeStampClass,
+    'com.linkedin.pegasus2avro.common.VersionProperties': VersionPropertiesClass,
     'com.linkedin.pegasus2avro.common.VersionTag': VersionTagClass,
     'com.linkedin.pegasus2avro.common.WindowDuration': WindowDurationClass,
     'com.linkedin.pegasus2avro.common.fieldtransformer.TransformationType': TransformationTypeClass,
@@ -24988,6 +25247,7 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.metadata.key.TagKey': TagKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.TelemetryKey': TelemetryKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.TestKey': TestKeyClass,
+    'com.linkedin.pegasus2avro.metadata.key.VersionSetKey': VersionSetKeyClass,
     'com.linkedin.pegasus2avro.metadata.query.filter.Condition': ConditionClass,
     'com.linkedin.pegasus2avro.metadata.query.filter.ConjunctiveCriterion': ConjunctiveCriterionClass,
     'com.linkedin.pegasus2avro.metadata.query.filter.Criterion': CriterionClass,
@@ -25159,6 +25419,8 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.usage.UsageAggregation': UsageAggregationClass,
     'com.linkedin.pegasus2avro.usage.UsageAggregationMetrics': UsageAggregationMetricsClass,
     'com.linkedin.pegasus2avro.usage.UserUsageCounts': UserUsageCountsClass,
+    'com.linkedin.pegasus2avro.versionset.VersionSetProperties': VersionSetPropertiesClass,
+    'com.linkedin.pegasus2avro.versionset.VersioningScheme': VersioningSchemeClass,
     'com.linkedin.pegasus2avro.view.DataHubViewDefinition': DataHubViewDefinitionClass,
     'com.linkedin.pegasus2avro.view.DataHubViewInfo': DataHubViewInfoClass,
     'com.linkedin.pegasus2avro.view.DataHubViewType': DataHubViewTypeClass,
@@ -25284,6 +25546,7 @@ __SCHEMA_TYPES = {
     'SubTypes': SubTypesClass,
     'TagAssociation': TagAssociationClass,
     'TimeStamp': TimeStampClass,
+    'VersionProperties': VersionPropertiesClass,
     'VersionTag': VersionTagClass,
     'WindowDuration': WindowDurationClass,
     'TransformationType': TransformationTypeClass,
@@ -25457,6 +25720,7 @@ __SCHEMA_TYPES = {
     'TagKey': TagKeyClass,
     'TelemetryKey': TelemetryKeyClass,
     'TestKey': TestKeyClass,
+    'VersionSetKey': VersionSetKeyClass,
     'Condition': ConditionClass,
     'ConjunctiveCriterion': ConjunctiveCriterionClass,
     'Criterion': CriterionClass,
@@ -25628,6 +25892,8 @@ __SCHEMA_TYPES = {
     'UsageAggregation': UsageAggregationClass,
     'UsageAggregationMetrics': UsageAggregationMetricsClass,
     'UserUsageCounts': UserUsageCountsClass,
+    'VersionSetProperties': VersionSetPropertiesClass,
+    'VersioningScheme': VersioningSchemeClass,
     'DataHubViewDefinition': DataHubViewDefinitionClass,
     'DataHubViewInfo': DataHubViewInfoClass,
     'DataHubViewType': DataHubViewTypeClass,
@@ -25659,6 +25925,7 @@ ASPECT_CLASSES: List[Type[_Aspect]] = [
     InputFieldsClass,
     OwnershipClass,
     DocumentationClass,
+    VersionPropertiesClass,
     BrowsePathsV2Class,
     OriginClass,
     AccessClass,
@@ -25744,6 +26011,7 @@ ASPECT_CLASSES: List[Type[_Aspect]] = [
     AssertionInfoClass,
     AssertionActionsClass,
     TelemetryClientIdClass,
+    VersionSetPropertiesClass,
     DataHubSecretValueClass,
     DataHubUpgradeRequestClass,
     DataHubUpgradeResultClass,
@@ -25788,6 +26056,7 @@ ASPECT_CLASSES: List[Type[_Aspect]] = [
     DataHubPersonaKeyClass,
     DataHubSecretKeyClass,
     DataHubIngestionSourceKeyClass,
+    VersionSetKeyClass,
     DataHubRetentionKeyClass,
     TestKeyClass,
     ChartKeyClass,
@@ -25874,6 +26143,7 @@ class AspectBag(TypedDict, total=False):
     inputFields: InputFieldsClass
     ownership: OwnershipClass
     documentation: DocumentationClass
+    versionProperties: VersionPropertiesClass
     browsePathsV2: BrowsePathsV2Class
     origin: OriginClass
     access: AccessClass
@@ -25959,6 +26229,7 @@ class AspectBag(TypedDict, total=False):
     assertionInfo: AssertionInfoClass
     assertionActions: AssertionActionsClass
     telemetryClientId: TelemetryClientIdClass
+    versionSetProperties: VersionSetPropertiesClass
     dataHubSecretValue: DataHubSecretValueClass
     dataHubUpgradeRequest: DataHubUpgradeRequestClass
     dataHubUpgradeResult: DataHubUpgradeResultClass
@@ -26003,6 +26274,7 @@ class AspectBag(TypedDict, total=False):
     dataHubPersonaKey: DataHubPersonaKeyClass
     dataHubSecretKey: DataHubSecretKeyClass
     dataHubIngestionSourceKey: DataHubIngestionSourceKeyClass
+    versionSetKey: VersionSetKeyClass
     dataHubRetentionKey: DataHubRetentionKeyClass
     testKey: TestKeyClass
     chartKey: ChartKeyClass
@@ -26106,6 +26378,7 @@ KEY_ASPECTS: Dict[str, Type[_Aspect]] = {
     'dataHubPersona': DataHubPersonaKeyClass,
     'dataHubSecret': DataHubSecretKeyClass,
     'dataHubIngestionSource': DataHubIngestionSourceKeyClass,
+    'versionSet': VersionSetKeyClass,
     'dataHubRetention': DataHubRetentionKeyClass,
     'test': TestKeyClass,
     'chart': ChartKeyClass,

datahub/metadata/_urns/urn_defs.py

@@ -2225,6 +2225,60 @@ class DataHubIngestionSourceUrn(_SpecificUrn):
     def id(self) -> str:
         return self.entity_ids[0]
 
+if TYPE_CHECKING:
+    from datahub.metadata.schema_classes import VersionSetKeyClass
+
+class VersionSetUrn(_SpecificUrn):
+    ENTITY_TYPE: ClassVar[str] = "versionSet"
+    _URN_PARTS: ClassVar[int] = 2
+
+    def __init__(self, id: str, entity_type: str, *, _allow_coercion: bool = True) -> None:
+        if _allow_coercion:
+            # Field coercion logic (if any is required).
+            id = UrnEncoder.encode_string(id)
+            entity_type = UrnEncoder.encode_string(entity_type)
+
+        # Validation logic.
+        if not id:
+            raise InvalidUrnError("VersionSetUrn id cannot be empty")
+        if UrnEncoder.contains_reserved_char(id):
+            raise InvalidUrnError(f'VersionSetUrn id contains reserved characters')
+        if not entity_type:
+            raise InvalidUrnError("VersionSetUrn entity_type cannot be empty")
+        if UrnEncoder.contains_reserved_char(entity_type):
+            raise InvalidUrnError(f'VersionSetUrn entity_type contains reserved characters')
+
+        super().__init__(self.ENTITY_TYPE, [id, entity_type])
+
+    @classmethod
+    def _parse_ids(cls, entity_ids: List[str]) -> "VersionSetUrn":
+        if len(entity_ids) != cls._URN_PARTS:
+            raise InvalidUrnError(f"VersionSetUrn should have {cls._URN_PARTS} parts, got {len(entity_ids)}: {entity_ids}")
+        return cls(id=entity_ids[0], entity_type=entity_ids[1], _allow_coercion=False)
+
+    @classmethod
+    def underlying_key_aspect_type(cls) -> Type["VersionSetKeyClass"]:
+        from datahub.metadata.schema_classes import VersionSetKeyClass
+
+        return VersionSetKeyClass
+
+    def to_key_aspect(self) -> "VersionSetKeyClass":
+        from datahub.metadata.schema_classes import VersionSetKeyClass
+
+        return VersionSetKeyClass(id=self.id, entityType=self.entity_type)
+
+    @classmethod
+    def from_key_aspect(cls, key_aspect: "VersionSetKeyClass") -> "VersionSetUrn":
+        return cls(id=key_aspect.id, entity_type=key_aspect.entityType)
+
+    @property
+    def id(self) -> str:
+        return self.entity_ids[0]
+
+    @property
+    def entity_type(self) -> str:
+        return self.entity_ids[1]
+
 if TYPE_CHECKING:
     from datahub.metadata.schema_classes import DataHubRetentionKeyClass
 

datahub/metadata/com/linkedin/pegasus2avro/common/__init__.py

@@ -65,6 +65,7 @@ from .....schema_classes import StatusClass
 from .....schema_classes import SubTypesClass
 from .....schema_classes import TagAssociationClass
 from .....schema_classes import TimeStampClass
+from .....schema_classes import VersionPropertiesClass
 from .....schema_classes import VersionTagClass
 from .....schema_classes import WindowDurationClass
 
@@ -127,6 +128,7 @@ Status = StatusClass
 SubTypes = SubTypesClass
 TagAssociation = TagAssociationClass
 TimeStamp = TimeStampClass
+VersionProperties = VersionPropertiesClass
 VersionTag = VersionTagClass
 WindowDuration = WindowDurationClass