Skip to content

Commit 01182ee

Browse files
author
gdj0nes
committed
FIX: yapf
1 parent 1edc45a commit 01182ee

File tree

2 files changed

+13
-10
lines changed

2 files changed

+13
-10
lines changed

labelbox/schema/project.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -569,7 +569,10 @@ def setup(self, labeling_frontend, labeling_frontend_options) -> None:
569569
timestamp = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
570570
self.update(setup_complete=timestamp)
571571

572-
def create_batch(self, name: str, data_rows: List[Union[str, "DataRow"]], priority: "BatchPriority" = 5):
572+
def create_batch(self,
573+
name: str,
574+
data_rows: List[Union[str, "DataRow"]],
575+
priority: "BatchPriority" = 5):
573576
"""Create a new batch for a project
574577
575578
Args:
@@ -595,8 +598,7 @@ def create_batch(self, name: str, data_rows: List[Union[str, "DataRow"]], priori
595598

596599
if len(dr_ids) > 25_000:
597600
raise ValueError(
598-
f"Batch exceeds max size, break into smaller batches"
599-
)
601+
f"Batch exceeds max size, break into smaller batches")
600602
if not len(dr_ids):
601603
raise ValueError("You need at least one data row in a batch")
602604

@@ -619,7 +621,8 @@ def create_batch(self, name: str, data_rows: List[Union[str, "DataRow"]], priori
619621
}
620622
}
621623

622-
res = self.client.execute(query_str, params, experimental=True)["project"][method]
624+
res = self.client.execute(query_str, params,
625+
experimental=True)["project"][method]
623626
res['size'] = len(dr_ids)
624627
return Entity.Batch(self.client, res)
625628

@@ -939,7 +942,7 @@ class LabelingParameterOverride(DbObject):
939942

940943
LabelerPerformance = namedtuple(
941944
"LabelerPerformance", "user count seconds_per_label, total_time_labeling "
942-
"consensus average_benchmark_agreement last_activity_time")
945+
"consensus average_benchmark_agreement last_activity_time")
943946
LabelerPerformance.__doc__ = (
944947
"Named tuple containing info about a labeler's performance.")
945948

tests/integration/test_batch.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@
99
@pytest.fixture
1010
def big_dataset(dataset: Dataset):
1111
task = dataset.create_data_rows([
12-
{
13-
"row_data": IMAGE_URL,
14-
"external_id": "my-image"
15-
},
16-
] * 250)
12+
{
13+
"row_data": IMAGE_URL,
14+
"external_id": "my-image"
15+
},
16+
] * 250)
1717
task.wait_till_done()
1818

1919
yield dataset

0 commit comments

Comments
 (0)