Add model_setup_complete to Project

Author: Val Brodsky

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

@@ -13,13 +13,9 @@
 
 from labelbox import parser
 from labelbox import utils
-from labelbox.exceptions import (
-    InvalidQueryError,
-    LabelboxError,
-    ProcessingWaitTimeout,
-    ResourceConflict,
-    ResourceNotFoundError
-)
+from labelbox.exceptions import (InvalidQueryError, LabelboxError,
+                                 ProcessingWaitTimeout, ResourceConflict,
+                                 ResourceNotFoundError)
 from labelbox.orm import query
 from labelbox.orm.db_object import DbObject, Deletable, Updateable, experimental
 from labelbox.orm.model import Entity, Field, Relationship
@@ -121,7 +117,11 @@ 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)
 
     # Relationships
     created_by = Relationship.ToOne("User", False, "created_by")
@@ -1752,7 +1752,9 @@ def __check_data_rows_have_been_processed(
         return response["queryAllDataRowsHaveBeenProcessed"][
             "allDataRowsHaveBeenProcessed"]
 
-    def get_overview(self, details=False) -> Union[ProjectOverview, ProjectOverviewDetailed]:
+    def get_overview(
+            self,
+            details=False) -> Union[ProjectOverview, ProjectOverviewDetailed]:
         """Return the overview of a project.
 
         This method returns the number of data rows per task queue and issues of a project,
@@ -1792,7 +1794,7 @@ def get_overview(self, details=False) -> Union[ProjectOverview, ProjectOverviewD
 
         # Must use experimental to access "issues"
         result = self.client.execute(query, {"projectId": self.uid},
-                                        experimental=True)["project"]
+                                     experimental=True)["project"]
 
         # Reformat category names
         overview = {
@@ -1805,26 +1807,28 @@ def get_overview(self, details=False) -> Union[ProjectOverview, ProjectOverviewD
 
         # Rename categories
         overview["to_label"] = overview.pop("unlabeled")
-        overview["total_data_rows"] = overview.pop("all")        
+        overview["total_data_rows"] = overview.pop("all")
 
         if not details:
             return ProjectOverview(**overview)
         else:
             # Build dictionary for queue details for review and rework queues
             for category in ["rework", "review"]:
                 queues = [
-                    {tq["name"]: tq.get("dataRowCount")}
+                    {
+                        tq["name"]: tq.get("dataRowCount")
+                    }
                     for tq in result.get("taskQueues")
                     if tq.get("queueType") == f"MANUAL_{category.upper()}_QUEUE"
                 ]
 
-                overview[f"in_{category}"]  = {
+                overview[f"in_{category}"] = {
                     "data": queues,
                     "total": overview[f"in_{category}"]
                 }
-            
+
             return ProjectOverviewDetailed(**overview)
-    
+
     def clone(self) -> "Project":
         """
         Clones the current project.

libs/labelbox/tests/integration/test_chat_evaluation_ontology_project.py

@@ -24,6 +24,8 @@ def test_create_chat_evaluation_ontology_project(
         assert classification.feature_schema_id
 
     project = chat_evaluation_project_create_dataset
+    assert project.model_setup_complete is None
+
     project.setup_editor(ontology)
 
     assert project.labeling_frontend().name == "Editor"