From 3001c8f6edcaa09839b31ef248f056735827eca0 Mon Sep 17 00:00:00 2001 From: "He Huang (Steve)" <105218074+stevehuang52@users.noreply.github.com> Date: Thu, 27 Jul 2023 08:22:37 -0400 Subject: [PATCH] Fix plot function in vad_utils.py (#7113) Fix plot function in vad_utils.py Signed-off-by: He Huang (Steve) <105218074+stevehuang52@users.noreply.github.com> --- nemo/collections/asr/parts/utils/vad_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nemo/collections/asr/parts/utils/vad_utils.py b/nemo/collections/asr/parts/utils/vad_utils.py index 44f2abec584f..fb511aacd816 100644 --- a/nemo/collections/asr/parts/utils/vad_utils.py +++ b/nemo/collections/asr/parts/utils/vad_utils.py @@ -1051,11 +1051,11 @@ def plot( else: label = None - if label: + if label is not None: ax2.plot(np.arange(len_pred) * unit_frame_len, label, 'r', label='label') - if pred_snippet: + if pred_snippet is not None: ax2.plot(np.arange(len_pred) * unit_frame_len, pred_snippet, 'b', label='pred') - if frame_snippet: + if frame_snippet is not None: ax2.plot(np.arange(len_pred) * unit_frame_len, frame_snippet, 'g--', label='speech prob') ax2.tick_params(axis='y', labelcolor='r')