Run batch from form

/**
 * Import categories batch.
 */
public static function importCategories($arg1, $arg2, &$context) {
 $a = 1;
}
/**
 * {@inheritdoc}
 */
public function submitForm(array &$form, FormStateInterface $form_state) {
 $data_to_process = [
   ['foo'],
   ['baz'],
 ];
 $batch = new BatchBuilder();
 $batch
   ->setTitle($this->t('Import categories'))
   ->setInitMessage($this->t('Prepare operations for processing.'));
 foreach ($data_to_process as $data) {
   $batch->addOperation([$this, 'importCategories'], $data);
 }
 batch_set($batch->toArray());
}