kovacsvi commited on
Commit
6577743
·
1 Parent(s): 658551c

sort is in-place

Browse files
Files changed (1) hide show
  1. interfaces/cap_minor_media.py +2 -1
interfaces/cap_minor_media.py CHANGED
@@ -42,7 +42,8 @@ major_to_minor_map = dict(major_to_minor_map)
42
  def normalize_probs(probs: dict, n: int):
43
  probs = list(probs.values())
44
  if len(probs) > n:
45
- probs = probs.sort(reverse=True)[:5]
 
46
  values = np.array(probs)
47
  exp_values = np.exp(values)
48
  sum_exp = np.sum(exp_values)
 
42
  def normalize_probs(probs: dict, n: int):
43
  probs = list(probs.values())
44
  if len(probs) > n:
45
+ probs.sort(reverse=True)
46
+ probs = probs[:5]
47
  values = np.array(probs)
48
  exp_values = np.exp(values)
49
  sum_exp = np.sum(exp_values)