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')