model: transformer: qa: Raise on invalid local rank

Coverity: 294618

Signed-off-by: John Andersen <[email protected]>

Author: pdxjohnny

model/transformers/dffml_model_transformers/qa/qa_model.py

@@ -236,6 +236,10 @@ async def _custom_train(self, train_dataset):
         """ Train the model """
         if self.parent.config.local_rank in [-1, 0]:
             tb_writer = SummaryWriter(log_dir=self.parent.config.log_dir)
+        else:
+            raise NotImplementedError(
+                f"local_rank was not -1 or 0: {self.parent.config.local_rank}"
+            )
 
         self.parent.config.train_batch_size = (
             self.parent.config.per_gpu_train_batch_size