Fix plot function in vad_utils.py (NVIDIA#7113)

Fix plot function in vad_utils.py Signed-off-by: He Huang (Steve) <[email protected]>
rohitrango · Jul 27, 2023 · f1ed44b · f1ed44b
1 parent 9f6e787
commit f1ed44b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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')