@@ -117,11 +117,8 @@ class Project(DbObject, Updateable, Deletable):
117
117
# Bind data_type and allowedMediaTYpe using the GraphQL type MediaType
118
118
media_type = Field .Enum (MediaType , "media_type" , "allowedMediaType" )
119
119
editor_task_type = Field .Enum (EditorTaskType , "editor_task_type" )
120
- < << << << HEAD
121
120
data_row_count = Field .Int ("data_row_count" )
122
- == == == =
123
- model_setup_complete = Field .Boolean ("model_setup_complete" )
124
- >> >> >> > 8 d9b99f7 (Add model_setup_complete to Project )
121
+ model_setup_complete : Field = Field .Boolean ("model_setup_complete" )
125
122
126
123
# Relationships
127
124
created_by = Relationship .ToOne ("User" , False , "created_by" )
@@ -1299,6 +1296,24 @@ def delete_project_model_config(self, project_model_config_id: str) -> bool:
1299
1296
raise ResourceNotFoundError (ProjectModelConfig , params )
1300
1297
return result ["deleteProjectModelConfig" ]["success" ]
1301
1298
1299
+ def set_project_model_setup_complete (self ) -> bool :
1300
+ """ Checks if the model setup is complete for this project.
1301
+
1302
+ Returns:
1303
+ bool, indicates if the model setup is complete.
1304
+ """
1305
+ query = """query ModelSetupCompletePyApi($projectId: ID!) {
1306
+ project(where: {id: $projectId}, data: {modelSetupComplete: true}) {
1307
+ modelSetupComplete
1308
+ }
1309
+ }"""
1310
+
1311
+ result = self .client .execute (query , {"projectId" : self .uid })
1312
+
1313
+ self .update (
1314
+ model_setup_complete = result ["project" ]["modelSetupComplete" ])
1315
+ return result ["project" ]["modelSetupComplete" ]
1316
+
1302
1317
def set_labeling_parameter_overrides (
1303
1318
self , data : List [LabelingParameterOverrideInput ]) -> bool :
1304
1319
""" Adds labeling parameter overrides to this project.
0 commit comments