Use _CoreProjectInput in create_offline_model_evaluation_project

Author: Val Brodsky

libs/labelbox/src/labelbox/client.py

@@ -604,9 +604,13 @@ def create_project(
         description: Optional[str] = None,
         auto_audit_percentage: Optional[float] = None,
         auto_audit_number_of_labels: Optional[int] = None,
-        quality_modes: Optional[Set[QualityMode]] = {QualityMode.Benchmark, QualityMode.Consensus},
+        quality_modes: Optional[Set[QualityMode]] = {
+            QualityMode.Benchmark,
+            QualityMode.Consensus,
+        },
         is_benchmark_enabled: Optional[bool] = None,
-        is_consensus_enabled: Optional[bool] = None) -> Project:
+        is_consensus_enabled: Optional[bool] = None,
+    ) -> Project:
         """Creates a Project object on the server.
 
         Attribute values are passed as keyword arguments.
@@ -655,7 +659,10 @@ def create_model_evaluation_project(
         description: Optional[str] = None,
         auto_audit_percentage: Optional[float] = None,
         auto_audit_number_of_labels: Optional[int] = None,
-        quality_modes: Optional[Set[QualityMode]] = {QualityMode.Benchmark, QualityMode.Consensus},
+        quality_modes: Optional[Set[QualityMode]] = {
+            QualityMode.Benchmark,
+            QualityMode.Consensus,
+        },
         is_benchmark_enabled: Optional[bool] = None,
         is_consensus_enabled: Optional[bool] = None,
         dataset_id: Optional[str] = None,
@@ -718,27 +725,41 @@ def create_model_evaluation_project(
         }
         return self._create_project(_CoreProjectInput(**input))
 
-    def create_offline_model_evaluation_project(self, **kwargs) -> Project:
+    def create_offline_model_evaluation_project(
+        self,
+        name: str,
+        description: Optional[str] = None,
+        auto_audit_percentage: Optional[float] = None,
+        auto_audit_number_of_labels: Optional[int] = None,
+        quality_modes: Optional[Set[QualityMode]] = {
+            QualityMode.Benchmark,
+            QualityMode.Consensus,
+        },
+        is_benchmark_enabled: Optional[bool] = None,
+        is_consensus_enabled: Optional[bool] = None,
+    ) -> Project:
         """
         Creates a project for offline model evaluation.
         Args:
             **kwargs: Additional parameters to pass see the create_project method
         Returns:
             Project: The created project
         """
-        kwargs["media_type"] = (
-            MediaType.Conversational
-        )  # Only Conversational is supported
-        kwargs["editor_task_type"] = (
-            EditorTaskType.OfflineModelChatEvaluation.value
-        )  # Special editor task type for offline model evaluation
-
-        # The following arguments are not supported for offline model evaluation
-        kwargs.pop("dataset_name_or_id", None)
-        kwargs.pop("append_to_existing_dataset", None)
-        kwargs.pop("data_row_count", None)
+        media_type = MediaType.Conversational
+        editor_task_type = EditorTaskType.OfflineModelChatEvaluation
 
-        return self._create_project(_CoreProjectInput(**kwargs))
+        input = {
+            "name": name,
+            "description": description,
+            "media_type": media_type,
+            "auto_audit_percentage": auto_audit_percentage,
+            "auto_audit_number_of_labels": auto_audit_number_of_labels,
+            "quality_modes": quality_modes,
+            "is_benchmark_enabled": is_benchmark_enabled,
+            "is_consensus_enabled": is_consensus_enabled,
+            "editor_task_type": editor_task_type,
+        }
+        return self._create_project(_CoreProjectInput(**input))
 
     def create_prompt_response_generation_project(
         self,