Skip to content

Commit 88ba72e

Browse files
author
Val Brodsky
committed
Add Project set_project_model_setup_complete
1 parent b497afb commit 88ba72e

File tree

1 file changed

+19
-4
lines changed

1 file changed

+19
-4
lines changed

libs/labelbox/src/labelbox/schema/project.py

Lines changed: 19 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -117,11 +117,8 @@ class Project(DbObject, Updateable, Deletable):
117117
# Bind data_type and allowedMediaTYpe using the GraphQL type MediaType
118118
media_type = Field.Enum(MediaType, "media_type", "allowedMediaType")
119119
editor_task_type = Field.Enum(EditorTaskType, "editor_task_type")
120-
<<<<<<< HEAD
121120
data_row_count = Field.Int("data_row_count")
122-
=======
123-
model_setup_complete = Field.Boolean("model_setup_complete")
124-
>>>>>>> 8d9b99f7 (Add model_setup_complete to Project)
121+
model_setup_complete: Field = Field.Boolean("model_setup_complete")
125122

126123
# Relationships
127124
created_by = Relationship.ToOne("User", False, "created_by")
@@ -1299,6 +1296,24 @@ def delete_project_model_config(self, project_model_config_id: str) -> bool:
12991296
raise ResourceNotFoundError(ProjectModelConfig, params)
13001297
return result["deleteProjectModelConfig"]["success"]
13011298

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+
13021317
def set_labeling_parameter_overrides(
13031318
self, data: List[LabelingParameterOverrideInput]) -> bool:
13041319
""" Adds labeling parameter overrides to this project.

0 commit comments

Comments
 (0)