@@ -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 )
0 commit comments