fixing score

ACEnglish · Jul 19, 2023 · 72a6d16 · 72a6d16
1 parent a8f2254
commit 72a6d16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utmos/select.py b/utmos/select.py
@@ -35,7 +35,7 @@ def do_sum(matrix, sample_mask):
         m_score += row
         m_count += (row != 0).astype('int')
     # mask out excluded/used samples
-    m_count[sample_mask != 1] = 0
+    m_score[sample_mask != 1] = 0
     return m_score, m_count