chore: rename variables in lossless coding submodule

src/komm/_lossless_coding/FanoCode.py

@@ -66,17 +66,17 @@ def fano_algorithm(pmf: PMF, source_block_size: int) -> dict[Word, Word]:
     xpmf = extended_probabilities(pmf, source_block_size, pbar)
     stack: list[tuple[list[tuple[Word, float]], Word]] = [(xpmf, ())]
     while stack:
-        current_pmf, prefix = stack.pop()
-        if len(current_pmf) == 1:
-            u, _ = current_pmf[0]
-            enc_mapping[u] = prefix
+        group, v = stack.pop()
+        if len(group) == 1:
+            u, _ = group[0]
+            enc_mapping[u] = v
             pbar.update()
             continue
-        probs = [p for _, p in current_pmf]
+        probs = [p for _, p in group]
         total = np.sum(probs)
         index = np.argmin(np.abs(np.cumsum(probs) - total / 2))
-        stack.append((current_pmf[index + 1 :], prefix + (1,)))
-        stack.append((current_pmf[: index + 1], prefix + (0,)))
+        stack.append((group[index + 1 :], v + (1,)))
+        stack.append((group[: index + 1], v + (0,)))
 
     pbar.close()
 

src/komm/_lossless_coding/util.py

@@ -137,7 +137,7 @@ def extended_probabilities(pmf: PMF, k: int, pbar: Any) -> list[tuple[Word, floa
         probs.append((-pu, u))
         if pbar is not None:
             pbar.update()
-    return [(u, -p) for p, u in sorted(probs)]
+    return [(u, -minus_pu) for minus_pu, u in sorted(probs)]
 
 
 def empty_mapping(cardinality: int, block_size: int) -> dict[Word, Word]: