czyoung commited on
Commit
f8088cc
·
verified ·
1 Parent(s): c4a0bf3

Fix 'group' as single voice bug

Browse files
Files changed (1) hide show
  1. sonogram.py +2 -2
sonogram.py CHANGED
@@ -199,7 +199,7 @@ class Sonogram():
199
  individuals.add(sas[j])
200
  if silenceCount > windowSize / 2:
201
  decision = 'silence'
202
- elif groupCount > windowSize / 2 or len(individuals) > 2:
203
  decision = 'group'
204
  else:
205
  decision = 'individual'
@@ -217,7 +217,7 @@ class Sonogram():
217
  if cat == 'group':
218
  timeStepMembers[i] = '+'.join(timeStepMembers[i])
219
  elif cat == 'individual':
220
- timeStepMembers[i] = timeStepMembers[i][0]
221
  else:
222
  timeStepMembers[i] = None
223
 
 
199
  individuals.add(sas[j])
200
  if silenceCount > windowSize / 2:
201
  decision = 'silence'
202
+ elif sas[i] == 'group' or groupCount > windowSize / 2 or len(individuals) > 2:
203
  decision = 'group'
204
  else:
205
  decision = 'individual'
 
217
  if cat == 'group':
218
  timeStepMembers[i] = '+'.join(timeStepMembers[i])
219
  elif cat == 'individual':
220
+ timeStepMembers[i] = sas[i]
221
  else:
222
  timeStepMembers[i] = None
223