simplify traverse() in huffman_code()

adr26 · May 11, 2021 · ce09746 · ce09746
1 parent 47021c4
commit ce09746
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bitarray/util.py b/bitarray/util.py
@@ -379,9 +379,9 @@ def huff_tree(freq_map):
     result = {}
 
     def traverse(nd, prefix=bitarray(0, endian)):
-        if hasattr(nd, 'symbol'):  # leaf
+        try:                    # leaf
             result[nd.symbol] = prefix
-        else:  # parent, so traverse each of the children
+        except AttributeError:  # parent, so traverse each of the children
             traverse(nd.child[0], prefix + bitarray([0]))
             traverse(nd.child[1], prefix + bitarray([1]))