Skip to content

Commit 7110a20

Browse files
authored
Add explicit matplotlib dep for parakeet and sortformer (pytorch#19231)
### Summary The Metal, MLX, and CUDA Windows export workflows started failing on 2026-04-30 with `ModuleNotFoundError: No module named 'matplotlib'` when the parakeet/sortformer export scripts import `nemo.collections.asr`. NeMo's clustering_diarizer module top-level imports matplotlib via vad_utils, but matplotlib is not declared by `nemo_toolkit[asr]`. A release in some transitive dep stopped pulling matplotlib in incidentally. nemo_toolkit itself was unchanged (2.7.3 since 2026-04-23), so adding matplotlib explicitly is the targeted fix. Authored with Claude Code. ### Test plan CI cc @metascroy
1 parent 662a3f5 commit 7110a20

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
1-
nemo_toolkit[asr]
1+
nemo_toolkit[asr]==2.7.3
2+
matplotlib
Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
1-
nemo_toolkit[asr]
1+
nemo_toolkit[asr]==2.7.3
2+
matplotlib

0 commit comments

Comments
 (0)