Minor fixes

k2-fsa · Oct 24, 2024 · dccf714 · dccf714
1 parent aa6108c
commit dccf714
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/sherpa-onnx/csrc/offline-speaker-diarization-pyannote-impl.h b/sherpa-onnx/csrc/offline-speaker-diarization-pyannote-impl.h
@@ -503,6 +503,11 @@ class OfflineSpeakerDiarizationPyannoteImpl
       }
     }
 
+    if (k != cur_row_index) {
+      auto seq = Eigen::seqN(0, cur_row_index);
+      ans = ans(seq, Eigen::all);
+    }
+
     return ans;
   }