Merge pull request #639 from atlanhq/bump-to-release-6.1.1

[release] Bump to release `6.1.1`

Author: Aryamanz29

HISTORY.md

@@ -1,3 +1,19 @@
+## 6.1.1 (May 21, 2025)
+
+### New Features
+
+- Added a utility method (`AssetClient.process_assets()`) to simplify processing (e.g: updating) assets while iterating through search results.
+
+### Bug Fixes
+
+- Fixed an issue where the search pagination loop would break due to invalid assets.
+- If `typeName` is not explicitly provided in the search request, the SDK now defaults to retrieving only assets with the `Referenceable` supertype. This avoids including non-asset records in the response.
+
+### QOL Improvements
+
+- Regenerated the latest typedef models.
+- Added a retry loop to ensure the token is fully active before retrying the original request after a `401` response.
+
 ## 6.1.0 (May 13, 2025)
 
 ### New Features

pyatlan/model/assets/__init__.py

@@ -76,8 +76,8 @@
         "SnowflakeStage",
         "SnowflakeStream",
         "DatabricksUnityCatalogTag",
-        "CalculationView",
         "Database",
+        "CalculationView",
         "Procedure",
         "SnowflakeTag",
         "MatillionGroup",

pyatlan/model/assets/__init__.pyi

@@ -73,8 +73,8 @@ __all__ = [
     "SnowflakeStage",
     "SnowflakeStream",
     "DatabricksUnityCatalogTag",
-    "CalculationView",
     "Database",
+    "CalculationView",
     "Procedure",
     "SnowflakeTag",
     "MatillionGroup",

pyatlan/model/assets/auth_service.py

@@ -39,6 +39,10 @@ def __setattr__(self, name, value):
     """
     TBC
     """
+    ABAC_SERVICE: ClassVar[KeywordField] = KeywordField("abacService", "abacService")
+    """
+    TBC
+    """
     AUTH_SERVICE_IS_ENABLED: ClassVar[BooleanField] = BooleanField(
         "authServiceIsEnabled", "authServiceIsEnabled"
     )
@@ -61,6 +65,7 @@ def __setattr__(self, name, value):
     _convenience_properties: ClassVar[List[str]] = [
         "auth_service_type",
         "tag_service",
+        "abac_service",
         "auth_service_is_enabled",
         "auth_service_config",
         "auth_service_policy_last_sync",
@@ -86,6 +91,16 @@ def tag_service(self, tag_service: Optional[str]):
             self.attributes = self.Attributes()
         self.attributes.tag_service = tag_service
 
+    @property
+    def abac_service(self) -> Optional[str]:
+        return None if self.attributes is None else self.attributes.abac_service
+
+    @abac_service.setter
+    def abac_service(self, abac_service: Optional[str]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.abac_service = abac_service
+
     @property
     def auth_service_is_enabled(self) -> Optional[bool]:
         return (
@@ -127,6 +142,7 @@ def auth_service_policy_last_sync(
     class Attributes(Asset.Attributes):
         auth_service_type: Optional[str] = Field(default=None, description="")
         tag_service: Optional[str] = Field(default=None, description="")
+        abac_service: Optional[str] = Field(default=None, description="")
         auth_service_is_enabled: Optional[bool] = Field(default=None, description="")
         auth_service_config: Optional[Dict[str, str]] = Field(
             default=None, description=""

pyatlan/model/assets/core/__init__.py

@@ -190,8 +190,8 @@
 SnowflakeStage.Attributes.update_forward_refs(**localns)
 SnowflakeStream.Attributes.update_forward_refs(**localns)
 DatabricksUnityCatalogTag.Attributes.update_forward_refs(**localns)
-CalculationView.Attributes.update_forward_refs(**localns)
 Database.Attributes.update_forward_refs(**localns)
+CalculationView.Attributes.update_forward_refs(**localns)
 Procedure.Attributes.update_forward_refs(**localns)
 SnowflakeTag.Attributes.update_forward_refs(**localns)
 MatillionGroup.Attributes.update_forward_refs(**localns)

pyatlan/model/assets/core/auth_policy.py

@@ -15,6 +15,7 @@
     KeywordField,
     NumericField,
     RelationField,
+    TextField,
 )
 from pyatlan.model.structs import AuthPolicyCondition, AuthPolicyValiditySchedule
 from pyatlan.utils import init_guid, validate_required_fields
@@ -80,6 +81,12 @@ def __setattr__(self, name, value):
             return object.__setattr__(self, name, value)
         super().__setattr__(name, value)
 
+    POLICY_FILTER_CRITERIA: ClassVar[TextField] = TextField(
+        "policyFilterCriteria", "policyFilterCriteria"
+    )
+    """
+    TBC
+    """
     POLICY_TYPE: ClassVar[KeywordField] = KeywordField("policyType", "policyType")
     """
     TBC
@@ -181,6 +188,7 @@ def __setattr__(self, name, value):
     """
 
     _convenience_properties: ClassVar[List[str]] = [
+        "policy_filter_criteria",
         "policy_type",
         "policy_service_name",
         "policy_category",
@@ -201,6 +209,18 @@ def __setattr__(self, name, value):
         "access_control",
     ]
 
+    @property
+    def policy_filter_criteria(self) -> Optional[str]:
+        return (
+            None if self.attributes is None else self.attributes.policy_filter_criteria
+        )
+
+    @policy_filter_criteria.setter
+    def policy_filter_criteria(self, policy_filter_criteria: Optional[str]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.policy_filter_criteria = policy_filter_criteria
+
     @property
     def policy_type(self) -> Optional[AuthPolicyType]:
         return None if self.attributes is None else self.attributes.policy_type
@@ -398,6 +418,7 @@ def access_control(self, access_control: Optional[AccessControl]):
         self.attributes.access_control = access_control
 
     class Attributes(Asset.Attributes):
+        policy_filter_criteria: Optional[str] = Field(default=None, description="")
         policy_type: Optional[AuthPolicyType] = Field(default=None, description="")
         policy_service_name: Optional[str] = Field(default=None, description="")
         policy_category: Optional[str] = Field(default=None, description="")

pyatlan/model/assets/core/column.py

@@ -201,6 +201,18 @@ def __setattr__(self, name, value):
     """
     Sub-data type of this column.
     """
+    COLUMN_COMPRESSION: ClassVar[KeywordField] = KeywordField(
+        "columnCompression", "columnCompression"
+    )
+    """
+    Compression type of this column.
+    """
+    COLUMN_ENCODING: ClassVar[KeywordField] = KeywordField(
+        "columnEncoding", "columnEncoding"
+    )
+    """
+    Encoding type of this column.
+    """
     RAW_DATA_TYPE_DEFINITION: ClassVar[TextField] = TextField(
         "rawDataTypeDefinition", "rawDataTypeDefinition"
     )
@@ -548,6 +560,8 @@ def __setattr__(self, name, value):
     _convenience_properties: ClassVar[List[str]] = [
         "data_type",
         "sub_data_type",
+        "column_compression",
+        "column_encoding",
         "raw_data_type_definition",
         "order",
         "nested_column_order",
@@ -639,6 +653,26 @@ def sub_data_type(self, sub_data_type: Optional[str]):
             self.attributes = self.Attributes()
         self.attributes.sub_data_type = sub_data_type
 
+    @property
+    def column_compression(self) -> Optional[str]:
+        return None if self.attributes is None else self.attributes.column_compression
+
+    @column_compression.setter
+    def column_compression(self, column_compression: Optional[str]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.column_compression = column_compression
+
+    @property
+    def column_encoding(self) -> Optional[str]:
+        return None if self.attributes is None else self.attributes.column_encoding
+
+    @column_encoding.setter
+    def column_encoding(self, column_encoding: Optional[str]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.column_encoding = column_encoding
+
     @property
     def raw_data_type_definition(self) -> Optional[str]:
         return (
@@ -1452,6 +1486,8 @@ def table_partition(self, table_partition: Optional[TablePartition]):
     class Attributes(SQL.Attributes):
         data_type: Optional[str] = Field(default=None, description="")
         sub_data_type: Optional[str] = Field(default=None, description="")
+        column_compression: Optional[str] = Field(default=None, description="")
+        column_encoding: Optional[str] = Field(default=None, description="")
         raw_data_type_definition: Optional[str] = Field(default=None, description="")
         order: Optional[int] = Field(default=None, description="")
         nested_column_order: Optional[str] = Field(default=None, description="")