Skip to content

Commit ff42f17

Browse files
author
gdj0nes
committed
CHG: run yapf
1 parent 9b98910 commit ff42f17

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

labelbox/schema/project.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -616,14 +616,13 @@ def create_batch(self, name: str, data_rows: List[str], priority: int = 5):
616616
}
617617
}
618618

619-
res = self.client.execute(
620-
query_str, params, experimental=True
621-
)["project"][method]
619+
res = self.client.execute(query_str, params,
620+
experimental=True)["project"][method]
622621

623622
res['size'] = len(dr_ids)
624623
return Entity.Batch(self.client, res)
625624

626-
def _update_queue_mode(self, mode: QueueMode) -> QueueMode:
625+
def _update_queue_mode(self, mode: "Project.QueueMode") -> "Project.QueueMode":
627626

628627
if self.queue_mode() == mode:
629628
return mode
@@ -652,7 +651,7 @@ def _update_queue_mode(self, mode: QueueMode) -> QueueMode:
652651

653652
return mode
654653

655-
def queue_mode(self) -> QueueMode:
654+
def queue_mode(self) -> "Project.QueueMode":
656655
"""Provides the status of if queue mode is enabled in the project."""
657656

658657
query_str = """query %s($projectId: ID!) {
@@ -939,7 +938,7 @@ class LabelingParameterOverride(DbObject):
939938

940939
LabelerPerformance = namedtuple(
941940
"LabelerPerformance", "user count seconds_per_label, total_time_labeling "
942-
"consensus average_benchmark_agreement last_activity_time")
941+
"consensus average_benchmark_agreement last_activity_time")
943942
LabelerPerformance.__doc__ = (
944943
"Named tuple containing info about a labeler's performance.")
945944

tests/integration/test_project.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,7 @@ def test_queued_data_row_export(configured_project):
180180

181181

182182
def test_queue_mode(configured_project: Project):
183-
assert configured_project.queue_mode() == configured_project.QueueMode.Dataset
183+
assert configured_project.queue_mode(
184+
) == configured_project.QueueMode.Dataset
184185
configured_project.update(queue_mode=configured_project.QueueMode.Batch)
185186
assert configured_project.queue_mode() == configured_project.QueueMode.Batch

0 commit comments

Comments
 (0)