@@ -456,7 +456,8 @@ def create_batch(
456
456
project : str ,
457
457
batch_name : str ,
458
458
callback : str = "" ,
459
- instruction_batch : bool = False ,
459
+ calibration_batch : bool = False ,
460
+ self_label_batch : bool = False ,
460
461
) -> Batch :
461
462
"""Create a new Batch within a project.
462
463
https://docs.scale.com/reference#batch-creation
@@ -469,10 +470,14 @@ def create_batch(
469
470
callback (str, optional):
470
471
Email to notify, or URL to POST to
471
472
when a batch is complete.
472
- instruction_batch (bool):
473
+ calibration_batch (bool):
473
474
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.
476
481
477
482
Returns:
478
483
Batch: Created batch object
@@ -481,7 +486,8 @@ def create_batch(
481
486
payload = dict (
482
487
project = project ,
483
488
name = batch_name ,
484
- instruction_batch = instruction_batch ,
489
+ calibration_batch = calibration_batch ,
490
+ self_label_batch = self_label_batch ,
485
491
callback = callback ,
486
492
)
487
493
batchdata = self .api .post_request (endpoint , body = payload )
0 commit comments