Skip to content

Commit 1e1d94d

Browse files
author
Val Brodsky
committed
Use _CoreProjectInput in all create_project methods
1 parent f26397c commit 1e1d94d

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

libs/labelbox/src/labelbox/client.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -633,8 +633,7 @@ def create_project(self, **kwargs) -> Project:
633633
kwargs.pop("append_to_existing_dataset", None)
634634
kwargs.pop("data_row_count", None)
635635
kwargs.pop("editor_task_type", None)
636-
input = _CoreProjectInput(**kwargs)
637-
return self._create_project(input)
636+
return self._create_project(_CoreProjectInput(**kwargs))
638637

639638
@overload
640639
def create_model_evaluation_project(
@@ -708,7 +707,7 @@ def create_model_evaluation_project(
708707
kwargs["data_row_count"] = data_row_count
709708
kwargs["editor_task_type"] = EditorTaskType.ModelChatEvaluation.value
710709

711-
return self._create_project(**kwargs)
710+
return self._create_project(_CoreProjectInput(**kwargs))
712711

713712
def create_offline_model_evaluation_project(self, **kwargs) -> Project:
714713
"""
@@ -730,7 +729,7 @@ def create_offline_model_evaluation_project(self, **kwargs) -> Project:
730729
kwargs.pop("append_to_existing_dataset", None)
731730
kwargs.pop("data_row_count", None)
732731

733-
return self._create_project(**kwargs)
732+
return self._create_project(_CoreProjectInput(**kwargs))
734733

735734
def create_prompt_response_generation_project(
736735
self,
@@ -800,7 +799,7 @@ def create_prompt_response_generation_project(
800799

801800
kwargs.pop("editor_task_type", None)
802801

803-
return self._create_project(**kwargs)
802+
return self._create_project(_CoreProjectInput(**kwargs))
804803

805804
def create_response_creation_project(self, **kwargs) -> Project:
806805
"""
@@ -820,7 +819,7 @@ def create_response_creation_project(self, **kwargs) -> Project:
820819
kwargs.pop("append_to_existing_dataset", None)
821820
kwargs.pop("data_row_count", None)
822821

823-
return self._create_project(**kwargs)
822+
return self._create_project(_CoreProjectInput(**kwargs))
824823

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

0 commit comments

Comments
 (0)