Skip to content

Commit 5d004df

Browse files
add support rapid self_label_batch creation (#45)
1 parent 7ba150e commit 5d004df

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

scaleapi/__init__.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,8 @@ def create_batch(
456456
project: str,
457457
batch_name: str,
458458
callback: str = "",
459-
instruction_batch: bool = False,
459+
calibration_batch: bool = False,
460+
self_label_batch: bool = False,
460461
) -> Batch:
461462
"""Create a new Batch within a project.
462463
https://docs.scale.com/reference#batch-creation
@@ -469,10 +470,14 @@ def create_batch(
469470
callback (str, optional):
470471
Email to notify, or URL to POST to
471472
when a batch is complete.
472-
instruction_batch (bool):
473+
calibration_batch (bool):
473474
Only applicable for self serve projects.
474-
Create an instruction batch by setting
475-
the instruction_batch flag to true.
475+
Create a calibration_batch batch by setting
476+
the calibration_batch flag to true.
477+
self_label_batch (bool):
478+
Only applicable for self serve projects.
479+
Create a self_label batch by setting
480+
the self_label_batch flag to true.
476481
477482
Returns:
478483
Batch: Created batch object
@@ -481,7 +486,8 @@ def create_batch(
481486
payload = dict(
482487
project=project,
483488
name=batch_name,
484-
instruction_batch=instruction_batch,
489+
calibration_batch=calibration_batch,
490+
self_label_batch=self_label_batch,
485491
callback=callback,
486492
)
487493
batchdata = self.api.post_request(endpoint, body=payload)

scaleapi/_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
__version__ = "2.3.0"
1+
__version__ = "2.4.0"
22
__package_name__ = "scaleapi"

0 commit comments

Comments
 (0)