Sync Python SDK models from atlanhq/models

pyatlan/__init__.py

@@ -10,7 +10,9 @@
 
 _install_pydantic_v1_perf()
 
-from pyatlan.utils import REQUEST_ID_FILTER  # noqa: E402  (perf patch must install first)
+from pyatlan.utils import (
+    REQUEST_ID_FILTER,  # noqa: E402  (perf patch must install first)
+)
 
 # Version information
 try:

pyatlan/client/aio/app.py

@@ -43,9 +43,6 @@
     is_duplicate_name_conflict,
 )
 from pyatlan.errors import AtlanError, ErrorCode
-from pyatlan.model.apps import AppInput
-from pyatlan.model.assets import AppWorkflowRun
-from pyatlan.model.fluent_search import CompoundQuery, FluentSearch
 from pyatlan.model.app import (
     AppDeleteResponse,
     AppInfo,
@@ -61,7 +58,9 @@
     CreateApp,
     UpdateApp,
 )
-
+from pyatlan.model.apps import AppInput
+from pyatlan.model.assets import AppWorkflowRun
+from pyatlan.model.fluent_search import CompoundQuery, FluentSearch
 
 LOGGER = logging.getLogger(__name__)
 

pyatlan/client/aio/approval_workflow.py

@@ -13,11 +13,11 @@
     AsyncApiCaller,
 )
 from pyatlan.errors import ErrorCode, InvalidRequestError
-from pyatlan.model.enums import ApprovalWorkflowRequestType
 from pyatlan.model.approval_workflow import (
     ApprovalWorkflowBulkActionResponse,
     ApprovalWorkflowRequest,
 )
+from pyatlan.model.enums import ApprovalWorkflowRequestType
 
 
 def _raise_if_recipient_scoped(err: InvalidRequestError, group_key: str):

pyatlan/client/aio/client.py

@@ -34,6 +34,8 @@
     AsyncUserCache,
 )
 from pyatlan.client.aio.admin import AsyncAdminClient
+from pyatlan.client.aio.app import AsyncAppClient
+from pyatlan.client.aio.approval_workflow import AsyncApprovalWorkflowClient
 from pyatlan.client.aio.asset import AsyncAssetClient
 from pyatlan.client.aio.audit import AsyncAuditClient
 from pyatlan.client.aio.contract import AsyncContractClient
@@ -45,15 +47,13 @@
 from pyatlan.client.aio.oauth_client import AsyncOAuthClient
 from pyatlan.client.aio.open_lineage import AsyncOpenLineageClient
 from pyatlan.client.aio.query import AsyncQueryClient
+from pyatlan.client.aio.requests import AsyncRequestsClient
 from pyatlan.client.aio.role import AsyncRoleClient
 from pyatlan.client.aio.search_log import AsyncSearchLogClient
 from pyatlan.client.aio.sso import AsyncSSOClient
 from pyatlan.client.aio.task import AsyncTaskClient
-from pyatlan.client.aio.approval_workflow import AsyncApprovalWorkflowClient
-from pyatlan.client.aio.requests import AsyncRequestsClient
 from pyatlan.client.aio.token import AsyncTokenClient
 from pyatlan.client.aio.typedef import AsyncTypeDefClient
-from pyatlan.client.aio.app import AsyncAppClient
 from pyatlan.client.aio.user import AsyncUserClient
 from pyatlan.client.atlan import (
     CONNECTION_RETRY,

pyatlan/client/app.py

@@ -37,10 +37,6 @@
     is_duplicate_name_conflict,
 )
 from pyatlan.errors import AtlanError, ErrorCode
-from pyatlan.model.apps import AppInput
-from pyatlan.model.assets import AppWorkflowRun
-from pyatlan.model.enums import AppWorkflowRunStatus
-from pyatlan.model.fluent_search import CompoundQuery, FluentSearch
 from pyatlan.model.app import (
     AppDeleteResponse,
     AppInfo,
@@ -56,7 +52,10 @@
     CreateApp,
     UpdateApp,
 )
-
+from pyatlan.model.apps import AppInput
+from pyatlan.model.assets import AppWorkflowRun
+from pyatlan.model.enums import AppWorkflowRunStatus
+from pyatlan.model.fluent_search import CompoundQuery, FluentSearch
 
 LOGGER = logging.getLogger(__name__)
 

pyatlan/client/approval_workflow.py

@@ -13,11 +13,11 @@
     ApprovalWorkflowGetRequest,
 )
 from pyatlan.errors import ErrorCode, InvalidRequestError
-from pyatlan.model.enums import ApprovalWorkflowRequestType
 from pyatlan.model.approval_workflow import (
     ApprovalWorkflowBulkActionResponse,
     ApprovalWorkflowRequest,
 )
+from pyatlan.model.enums import ApprovalWorkflowRequestType
 
 
 def _raise_if_recipient_scoped(err: InvalidRequestError, group_key: str):

pyatlan/client/atlan.py

@@ -39,6 +39,8 @@
 from pyatlan.cache.source_tag_cache import SourceTagCache
 from pyatlan.cache.user_cache import UserCache
 from pyatlan.client.admin import AdminClient
+from pyatlan.client.app import AppClient
+from pyatlan.client.approval_workflow import ApprovalWorkflowClient
 from pyatlan.client.asset import A, AssetClient, IndexSearchResults, LineageListResults
 from pyatlan.client.audit import AuditClient
 from pyatlan.client.common import CONNECTION_RETRY, ImpersonateUser
@@ -52,7 +54,6 @@
 from pyatlan.client.oauth_client import OAuthClient
 from pyatlan.client.open_lineage import OpenLineageClient
 from pyatlan.client.query import QueryClient
-from pyatlan.client.approval_workflow import ApprovalWorkflowClient
 from pyatlan.client.requests import RequestsClient
 from pyatlan.client.role import RoleClient
 from pyatlan.client.search_log import SearchLogClient
@@ -61,7 +62,6 @@
 from pyatlan.client.token import TokenClient
 from pyatlan.client.transport import PyatlanSyncTransport  # type: ignore
 from pyatlan.client.typedef import TypeDefClient
-from pyatlan.client.app import AppClient
 from pyatlan.client.user import UserClient
 from pyatlan.errors import ERROR_CODE_FOR_HTTP_STATUS, AtlanError, ErrorCode
 from pyatlan.model.api_tokens import ApiToken, ApiTokenResponse

pyatlan/client/common/__init__.py

@@ -24,6 +24,12 @@
 # Admin shared logic classes
 from .admin import AdminGetAdminEvents, AdminGetKeycloakEvents
 
+# Role shared logic classes
+from .approval_workflow import (
+    ApprovalWorkflowBulkActionRequests,
+    ApprovalWorkflowGetRequest,
+)
+
 # Asset shared logic classes
 from .asset import (
     DeleteByGuid,
@@ -116,12 +122,6 @@
 
 # Query shared logic classes
 from .query import QueryStream
-
-# Role shared logic classes
-from .approval_workflow import (
-    ApprovalWorkflowBulkActionRequests,
-    ApprovalWorkflowGetRequest,
-)
 from .requests import (
     RequestsAction,
     RequestsCreate,

pyatlan/client/common/app.py

@@ -13,10 +13,9 @@
 import json
 from typing import Any, Dict, Optional, Tuple
 
-from pyatlan.errors import AtlanError
-
 from pyatlan.client.constants import (
     ADD_APP_SCHEDULE,
+    API,
     CANCEL_APP_RUN,
     CREATE_APP_WORKFLOW,
     DELETE_APP_WORKFLOW,
@@ -29,7 +28,7 @@
     SUBMIT_APP_WORKFLOW,
     UPDATE_APP_WORKFLOW,
 )
-from pyatlan.client.constants import API
+from pyatlan.errors import AtlanError
 from pyatlan.model.app import (
     AppDeleteResponse,
     AppInfo,

pyatlan/model/assets/bigquery_tag.py

@@ -1156,8 +1156,12 @@ class Attributes(Tag.Attributes):
 from .core.dbt_seed import DbtSeed  # noqa: E402, F401
 from .core.dbt_source import DbtSource  # noqa: E402, F401
 from .core.dbt_test import DbtTest  # noqa: E402, F401
-from .core.snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .core.sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .core.snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .core.sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .core.sql_insight_join import SqlInsightJoin  # noqa: E402, F401
 
 BigqueryTag.Attributes.update_forward_refs()

pyatlan/model/assets/core/cosmos_mongo_d_b_collection.py

@@ -2138,8 +2138,12 @@ class Attributes(CosmosMongoDB.Attributes):
 from .procedure import Procedure  # noqa: E402, F401
 from .query import Query  # noqa: E402, F401
 from .schema import Schema  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401
 from .table import Table  # noqa: E402, F401
 from .table_partition import TablePartition  # noqa: E402, F401

pyatlan/model/assets/core/cosmos_mongo_d_b_database.py

@@ -1289,6 +1289,10 @@ class Attributes(CosmosMongoDB.Attributes):
 from .fabric_workspace import FabricWorkspace  # noqa: E402, F401
 from .mongo_d_b_collection import MongoDBCollection  # noqa: E402, F401
 from .schema import Schema  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/databricks_a_i_model_context.py

@@ -1370,6 +1370,10 @@ class Attributes(AIModel.Attributes):
 from .dbt_source import DbtSource  # noqa: E402, F401
 from .dbt_test import DbtTest  # noqa: E402, F401
 from .schema import Schema  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/databricks_a_i_model_version.py

@@ -1670,6 +1670,10 @@ class Attributes(AIModelVersion.Attributes):
 from .dbt_seed import DbtSeed  # noqa: E402, F401
 from .dbt_source import DbtSource  # noqa: E402, F401
 from .dbt_test import DbtTest  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/databricks_genie_agent.py

@@ -1319,6 +1319,10 @@ class Attributes(Agent.Attributes):
 from .dbt_seed import DbtSeed  # noqa: E402, F401
 from .dbt_source import DbtSource  # noqa: E402, F401
 from .dbt_test import DbtTest  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/databricks_unity_catalog_tag.py

@@ -1086,6 +1086,10 @@ class Attributes(Tag.Attributes):
 from .dbt_seed import DbtSeed  # noqa: E402, F401
 from .dbt_source import DbtSource  # noqa: E402, F401
 from .dbt_test import DbtTest  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/document_d_b_database.py

@@ -1239,6 +1239,10 @@ def creator(
 from .document_d_b_collection import DocumentDBCollection  # noqa: E402, F401
 from .fabric_workspace import FabricWorkspace  # noqa: E402, F401
 from .schema import Schema  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/fabric_dataflow.py

@@ -91,5 +91,7 @@ class Attributes(Fabric.Attributes):
     )
 
 
-from .fabric_dataflow_entity_column import FabricDataflowEntityColumn  # noqa: E402, F401
+from .fabric_dataflow_entity_column import (
+    FabricDataflowEntityColumn,  # noqa: E402, F401
+)
 from .fabric_workspace import FabricWorkspace  # noqa: E402, F401

pyatlan/model/assets/core/fabric_semantic_model_table.py

@@ -132,4 +132,6 @@ class Attributes(Fabric.Attributes):
 
 
 from .fabric_semantic_model import FabricSemanticModel  # noqa: E402, F401
-from .fabric_semantic_model_table_column import FabricSemanticModelTableColumn  # noqa: E402, F401
+from .fabric_semantic_model_table_column import (
+    FabricSemanticModelTableColumn,  # noqa: E402, F401
+)

pyatlan/model/assets/core/power_b_i_dataflow.py

@@ -354,7 +354,9 @@ class Attributes(PowerBI.Attributes):
     )
 
 
-from .power_b_i_dataflow_entity_column import PowerBIDataflowEntityColumn  # noqa: E402, F401
+from .power_b_i_dataflow_entity_column import (
+    PowerBIDataflowEntityColumn,  # noqa: E402, F401
+)
 from .power_b_i_dataset import PowerBIDataset  # noqa: E402, F401
 from .power_b_i_datasource import PowerBIDatasource  # noqa: E402, F401
 from .power_b_i_table import PowerBITable  # noqa: E402, F401

pyatlan/model/assets/core/s_q_l.py

@@ -995,6 +995,10 @@ class Attributes(Catalog.Attributes):
 from .dbt_model import DbtModel  # noqa: E402, F401
 from .dbt_seed import DbtSeed  # noqa: E402, F401
 from .dbt_source import DbtSource  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/schema.py

@@ -621,7 +621,9 @@ def create(
 from .function import Function  # noqa: E402, F401
 from .materialised_view import MaterialisedView  # noqa: E402, F401
 from .procedure import Procedure  # noqa: E402, F401
-from .sap_datasphere_replication_flow import SapDatasphereReplicationFlow  # noqa: E402, F401
+from .sap_datasphere_replication_flow import (
+    SapDatasphereReplicationFlow,  # noqa: E402, F401
+)
 from .snowflake_a_i_model_context import SnowflakeAIModelContext  # noqa: E402, F401
 from .snowflake_pipe import SnowflakePipe  # noqa: E402, F401
 from .snowflake_semantic_view import SnowflakeSemanticView  # noqa: E402, F401

pyatlan/model/assets/core/snowflake_a_i_model_context.py

@@ -1340,6 +1340,10 @@ class Attributes(AIModel.Attributes):
 from .dbt_test import DbtTest  # noqa: E402, F401
 from .schema import Schema  # noqa: E402, F401
 from .snowflake_a_i_model_version import SnowflakeAIModelVersion  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/snowflake_a_i_model_version.py

@@ -1468,6 +1468,10 @@ class Attributes(AIModelVersion.Attributes):
 from .dbt_source import DbtSource  # noqa: E402, F401
 from .dbt_test import DbtTest  # noqa: E402, F401
 from .snowflake_a_i_model_context import SnowflakeAIModelContext  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/snowflake_semantic_dimension.py

@@ -1278,6 +1278,10 @@ class Attributes(SemanticDimension.Attributes):
 from .dbt_seed import DbtSeed  # noqa: E402, F401
 from .dbt_source import DbtSource  # noqa: E402, F401
 from .dbt_test import DbtTest  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401

pyatlan/model/assets/core/snowflake_semantic_fact.py

@@ -1072,4 +1072,6 @@ class Attributes(Snowflake.Attributes):
 
 
 from .semantic_model import SemanticModel  # noqa: E402, F401
-from .snowflake_semantic_logical_table import SnowflakeSemanticLogicalTable  # noqa: E402, F401
+from .snowflake_semantic_logical_table import (
+    SnowflakeSemanticLogicalTable,  # noqa: E402, F401
+)

pyatlan/model/assets/core/snowflake_semantic_logical_table.py

@@ -1407,5 +1407,7 @@ class Attributes(SemanticEntity.Attributes):
 from .snowflake_semantic_fact import SnowflakeSemanticFact  # noqa: E402, F401
 from .snowflake_semantic_metric import SnowflakeSemanticMetric  # noqa: E402, F401
 from .snowflake_semantic_view import SnowflakeSemanticView  # noqa: E402, F401
-from .sql_insight_business_question import SqlInsightBusinessQuestion  # noqa: E402, F401
+from .sql_insight_business_question import (
+    SqlInsightBusinessQuestion,  # noqa: E402, F401
+)
 from .sql_insight_join import SqlInsightJoin  # noqa: E402, F401