Update nemo/collections/asr/models/msdd_models.py

Signed-off-by: Sean Naren <[email protected]>
NVIDIA · Feb 15, 2023 · 5bf2273 · 5bf2273
1 parent 49b7bc6
commit 5bf2273
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/asr/models/msdd_models.py b/nemo/collections/asr/models/msdd_models.py
@@ -1190,7 +1190,7 @@ def diarize(self) -> Optional[List[Optional[List[Tuple[DiarizationErrorRate, Dic
         thresholds = list(self._cfg.diarizer.msdd_model.parameters.sigmoid_threshold)
         if self._cfg.evaluate:
             return [self.run_overlap_aware_eval(preds_list, threshold) for threshold in thresholds]
-        return
+        return None
 
     def get_range_average(
         self, signals: torch.Tensor, emb_vectors: torch.Tensor, diar_window_index: int, test_data_collection: List[Any]