Add Project set_project_model_setup_complete

Author: Val Brodsky

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

@@ -117,11 +117,8 @@ class Project(DbObject, Updateable, Deletable):
     # Bind data_type and allowedMediaTYpe using the GraphQL type MediaType
     media_type = Field.Enum(MediaType, "media_type", "allowedMediaType")
     editor_task_type = Field.Enum(EditorTaskType, "editor_task_type")
-<<<<<<< HEAD
     data_row_count = Field.Int("data_row_count")
-=======
-    model_setup_complete = Field.Boolean("model_setup_complete")
->>>>>>> 8d9b99f7 (Add model_setup_complete to Project)
+    model_setup_complete: Field = Field.Boolean("model_setup_complete")
 
     # Relationships
     created_by = Relationship.ToOne("User", False, "created_by")
@@ -1299,6 +1296,24 @@ def delete_project_model_config(self, project_model_config_id: str) -> bool:
             raise ResourceNotFoundError(ProjectModelConfig, params)
         return result["deleteProjectModelConfig"]["success"]
 
+    def set_project_model_setup_complete(self) -> bool:
+        """ Checks if the model setup is complete for this project.
+
+        Returns:
+            bool, indicates if the model setup is complete.
+        """
+        query = """query ModelSetupCompletePyApi($projectId: ID!) {
+            project(where: {id: $projectId}, data: {modelSetupComplete: true}) {
+                modelSetupComplete
+            }
+        }"""
+
+        result = self.client.execute(query, {"projectId": self.uid})
+
+        self.update(
+            model_setup_complete=result["project"]["modelSetupComplete"])
+        return result["project"]["modelSetupComplete"]
+
     def set_labeling_parameter_overrides(
             self, data: List[LabelingParameterOverrideInput]) -> bool:
         """ Adds labeling parameter overrides to this project.