Minor change in Sardinas–Patterson algorithm

rwnobrega · Jul 28, 2023 · 5739350 · 5739350
1 parent c8fc617
commit 5739350
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/komm/_source_coding/_util.py b/src/komm/_source_coding/_util.py
@@ -18,9 +18,9 @@ def is_uniquely_decodable(words: list[Word]) -> bool:
         for c1, c2 in it.product(augmented_words, repeat=2):
             if c1 != c2 and c1[: len(c2)] == c2:
                 dangling_suffixes.add(c1[len(c2) :])
-        if any(suffix in words for suffix in dangling_suffixes):
+        if dangling_suffixes & set(words):
             return False
-        if all(suffix in augmented_words for suffix in dangling_suffixes):
+        if dangling_suffixes <= augmented_words:
             return True
         augmented_words |= dangling_suffixes