Added processes_batch_save() method

Author: vaibhavatlan

pyatlan/model/assets/core/a_i_model.py

@@ -107,6 +107,13 @@ def processes_creator(
 
         return process_list
 
+    def processes_batch_save(client, process_list):
+        batch_size = 20
+        total_processes = len(process_list)
+        for i in range(0, total_processes, batch_size):
+            batch = process_list[i : i + batch_size]
+            client.asset.save(batch)
+
     type_name: str = Field(default="AIModel", allow_mutation=False)
 
     @validator("type_name")

tests/integration/ai_asset_test.py

@@ -145,7 +145,9 @@ def test_ai_model_processes_creator(
         AIDatasetType.OUTPUT: [Table.ref_by_guid(guid=guids[4])],
     }
     created_processes = AIModel.processes_creator(
-        a_i_model_guid=ai_model.guid, ai_model_name=ai_model.name, database_dict=database_dict
+        a_i_model_guid=ai_model.guid,
+        ai_model_name=ai_model.name,
+        database_dict=database_dict,
     )
 
     mutation_response = client.asset.save(created_processes)  # type: ignore