fix warning about potential uninitialized variable

Signed-off-by: Alexandra Antonova <[email protected]>
bene-ges · Dec 4, 2023 · 6507eba · 6507eba
1 parent e196bca
commit 6507eba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/nlp/models/spellchecking_asr_customization/spellchecking_model.py b/nemo/collections/nlp/models/spellchecking_asr_customization/spellchecking_model.py
@@ -288,7 +288,7 @@ def on_validation_epoch_end(self):
         split = "test" if self.trainer.testing else "val"
         if split == 'val':
             avg_loss = torch.stack([x[f'{split}_loss'] for x in self.validation_step_outputs]).mean()
-        elif split == 'test':
+        else:
             avg_loss = torch.stack([x[f'{split}_loss'] for x in self.test_step_outputs]).mean()
 
         # Calculate metrics and classification report