Add data_row_count to Project

Author: Val Brodsky

libs/labelbox/src/labelbox/client.py

@@ -821,7 +821,8 @@ def create_model_evaluation_project(self,
 
     def create_offline_model_evaluation_project(self, **kwargs) -> Project:
         kwargs["media_type"] = MediaType.Conversational
-        kwargs["editor_task_type"] = EditorTaskType.OfflineModelChatEvaluation.value
+        kwargs[
+            "editor_task_type"] = EditorTaskType.OfflineModelChatEvaluation.value
 
         return self.create_project(**kwargs)
 
@@ -900,8 +901,6 @@ def _create_project(self, **kwargs) -> Project:
                 params.pop("dataset_name_or_id", None),
             Field.Boolean("append_to_existing_dataset"):
                 params.pop("append_to_existing_dataset", None),
-            Field.Int("data_row_count"):
-                params.pop("data_row_count", None),
         }
         extra_params = {k: v for k, v in extra_params.items() if v is not None}
 

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

@@ -121,6 +121,7 @@ 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")
+    data_row_count = Field.Int("data_row_count")
 
     # Relationships
     created_by = Relationship.ToOne("User", False, "created_by")

libs/labelbox/tests/integration/conftest.py

@@ -425,6 +425,7 @@ def offline_chat_evaluation_project(client, rand_gen):
 
     project.delete()
 
+
 @pytest.fixture
 def chat_evaluation_project_append_to_dataset(client, dataset, rand_gen):
     project_name = f"test-model-evaluation-project-{rand_gen(str)}"
@@ -465,4 +466,4 @@ def print_perf_summary():
         num_of_entries = 10 if len(sorted_dict) >= 10 else len(sorted_dict)
         slowest_fixtures = [(aaa, sorted_dict[aaa])
                             for aaa in islice(sorted_dict, num_of_entries)]
-        print("\nTop slowest fixtures:\n", slowest_fixtures, file=sys.stderr)
+        print("\nTop slowest fixtures:\n", slowest_fixtures, file=sys.stderr)

libs/labelbox/tests/integration/test_offline_chat_evaluation_project.py

@@ -1,7 +1,6 @@
-def test_create_offline_chat_evaluation_project(
-        client,
-        offline_chat_evaluation_project,
-        chat_evaluation_ontology):
+def test_create_offline_chat_evaluation_project(client,
+                                                offline_chat_evaluation_project,
+                                                chat_evaluation_ontology):
     project = offline_chat_evaluation_project
     assert project