Skip to content

Commit 56b8bd5

Browse files
author
Val Brodsky
committed
Use _CoreProjectInput in create_response_creation_project
1 parent 1a63b4a commit 56b8bd5

File tree

1 file changed

+27
-17
lines changed

1 file changed

+27
-17
lines changed

libs/labelbox/src/labelbox/client.py

Lines changed: 27 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -745,19 +745,16 @@ def create_offline_model_evaluation_project(
745745
Returns:
746746
Project: The created project
747747
"""
748-
media_type = MediaType.Conversational
749-
editor_task_type = EditorTaskType.OfflineModelChatEvaluation
750-
751748
input = {
752749
"name": name,
753750
"description": description,
754-
"media_type": media_type,
751+
"media_type": MediaType.Conversational,
755752
"auto_audit_percentage": auto_audit_percentage,
756753
"auto_audit_number_of_labels": auto_audit_number_of_labels,
757754
"quality_modes": quality_modes,
758755
"is_benchmark_enabled": is_benchmark_enabled,
759756
"is_consensus_enabled": is_consensus_enabled,
760-
"editor_task_type": editor_task_type,
757+
"editor_task_type": EditorTaskType.OfflineModelChatEvaluation,
761758
}
762759
return self._create_project(_CoreProjectInput(**input))
763760

@@ -845,25 +842,38 @@ def create_prompt_response_generation_project(
845842
}
846843
return self._create_project(_CoreProjectInput(**input))
847844

848-
def create_response_creation_project(self, **kwargs) -> Project:
845+
def create_response_creation_project(
846+
self,
847+
name: str,
848+
description: Optional[str] = None,
849+
auto_audit_percentage: Optional[float] = None,
850+
auto_audit_number_of_labels: Optional[int] = None,
851+
quality_modes: Optional[Set[QualityMode]] = {
852+
QualityMode.Benchmark,
853+
QualityMode.Consensus,
854+
},
855+
is_benchmark_enabled: Optional[bool] = None,
856+
is_consensus_enabled: Optional[bool] = None,
857+
) -> Project:
849858
"""
850859
Creates a project for response creation.
851860
Args:
852861
**kwargs: Additional parameters to pass see the create_project method
853862
Returns:
854863
Project: The created project
855864
"""
856-
kwargs["media_type"] = MediaType.Text # Only Text is supported
857-
kwargs["editor_task_type"] = (
858-
EditorTaskType.ResponseCreation.value
859-
) # Special editor task type for response creation projects
860-
861-
# The following arguments are not supported for response creation projects
862-
kwargs.pop("dataset_name_or_id", None)
863-
kwargs.pop("append_to_existing_dataset", None)
864-
kwargs.pop("data_row_count", None)
865-
866-
return self._create_project(_CoreProjectInput(**kwargs))
865+
input = {
866+
"name": name,
867+
"description": description,
868+
"media_type": MediaType.Text, # Only Text is supported
869+
"auto_audit_percentage": auto_audit_percentage,
870+
"auto_audit_number_of_labels": auto_audit_number_of_labels,
871+
"quality_modes": quality_modes,
872+
"is_benchmark_enabled": is_benchmark_enabled,
873+
"is_consensus_enabled": is_consensus_enabled,
874+
"editor_task_type": EditorTaskType.ResponseCreation.value, # Special editor task type for response creation projects
875+
}
876+
return self._create_project(_CoreProjectInput(**input))
867877

868878
def _create_project(self, input: _CoreProjectInput) -> Project:
869879
media_type_value = input.media_type.value

0 commit comments

Comments
 (0)