AIModel

class pyatlan.model.assets.AIModel(*, typeName: str = 'AIModel', attributes: AIModel.Attributes = None, businessAttributes: Dict[str, Any] | None = None, createdBy: str | None = None, createTime: int | None = None, deleteHandler: str | None = None, guid: str = None, isIncomplete: bool | None = None, labels: List[str] | None = None, relationshipAttributes: RelationshipAttributes | Dict[str, Any] | None = None, status: EntityStatus | None = None, updatedBy: str | None = None, updateTime: int | None = None, version: int | None = None, classifications: List[AtlanTag] | None = None, classificationNames: List[str] | None = None, displayText: str | None = None, entityStatus: str | None = None, relationshipGuid: str | None = None, relationshipStatus: str | None = None, relationshipType: str | None = None, meaningNames: List[str] | None = None, meanings: List[Meaning] | None = None, customAttributes: Dict[str, Any] | None = None, scrubbed: bool | None = None, pendingTasks: List[str] | None = None, uniqueAttributes: Dict[str, Any] | None = None, appendRelationshipAttributes: Dict[str, Any] | None = None, removeRelationshipAttributes: Dict[str, Any] | None = None, addOrUpdateClassifications: List[AtlanTag] | None = None, removeClassifications: List[AtlanTag] | None = None, semantic: SaveSemantic | None = None, depth: int | None = None, immediateUpstream: List[LineageRef] | None = None, immediateDownstream: List[LineageRef] | None = None)

Description

AI_MODEL_DATASETS_DSL: ClassVar[TextField] = <pyatlan.model.fields.atlan_fields.TextField object>

Search DSL used to define which assets/datasets are part of the AI model.

AI_MODEL_STATUS: ClassVar[KeywordField] = <pyatlan.model.fields.atlan_fields.KeywordField object>

Status of the AI model

AI_MODEL_VERSION: ClassVar[KeywordField] = <pyatlan.model.fields.atlan_fields.KeywordField object>

Version of the AI model

AI_MODEL_VERSIONS: ClassVar[RelationField] = <pyatlan.model.fields.atlan_fields.RelationField object>

TBC

APPLICATIONS: ClassVar[RelationField] = <pyatlan.model.fields.atlan_fields.RelationField object>

TBC

classmethod processes_batch_save(client, process_list: List[Process]) → List

Saves a list of Process objects to Atlan in batches to optimize performance. We save the processes in batches of 20.

Parameters:

• client – Atlan client instance for making API calls

• process_list – list of Process objects to save

Returns:

list of API responses from each batch save operation

classmethod processes_creator(ai_model: AIModel, dataset_dict: Dict[AIDatasetType, list]) → List[Process]

Creates a list of Process objects representing the relationships between an AI model and its datasets.

Parameters:

• ai_model – the AI model for which to create processes

• dataset_dict – dictionary mapping AI dataset types to lists of assets

Returns:

list of Process objects representing the AI model’s data lineage

Raises:

ValueError – when the AI model is missing required attributes (guid or name)