oops

interfaces/cap_minor_media.py

@@ -40,11 +40,11 @@ for code in CAP_MIN_CODES:
 major_to_minor_map = dict(major_to_minor_map)
 
 def normalize_probs(probs: dict, n: int):
-    probs = list(probs.values())
-    if len(probs) > n:
-        probs.sort(reverse=True)
-        probs = probs[:5]
-    values = np.array(probs)
+    probs_ = list(probs.values())
+    if len(probs_) > n:
+        probs_.sort(reverse=True)
+        probs_ = probs[:5]
+    values = np.array(probs_)
     exp_values = np.exp(values)
     sum_exp = np.sum(exp_values)
     return {k: float(v) for k, v in zip(probs.keys(), exp_values / sum_exp)}