small cleanups

sourmash-bio · May 3, 2018 · 90b7b6b · 90b7b6b
1 parent 1de8717
commit 90b7b6b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/sourmash/commands.py b/sourmash/commands.py
@@ -710,7 +710,6 @@ def index(args):
         sys.exit(-1)
 
     notify('loaded {} sigs; saving SBT under "{}"', n, args.sbt_name)
-    tree._fill_internal()
     tree.save(args.sbt_name, sparseness=args.sparseness)
 
 

diff --git a/sourmash/sbt.py b/sourmash/sbt.py
@@ -388,7 +388,7 @@ def save(self, path, storage=None, sparseness=0.0):
 
         nodes = {}
         leaves = {}
-        total_nodes = len(self._nodes) + len(self._leaves)
+        total_nodes = len(self)
         for n, (i, node) in enumerate(self):
             if node is None:
                 continue
@@ -661,7 +661,7 @@ def fill_nodegraphs(node, *args, **kwargs):
         self.is_ready = True
 
     def _fill_up(self, search_fn, *args, **kwargs):
-        visited, queue = set(), list(reversed(list(self._leaves.keys())))
+        visited, queue = set(), list(reversed(sorted(self._leaves.keys())))
         while queue:
             node_p = queue.pop(0)
 
@@ -706,7 +706,7 @@ def print_dot(self):
         edge [arrowsize=0.8];
         """)
 
-        for i, node in list(self._nodes.items()):
+        for i, node in self._nodes.items():
             if isinstance(node, Node):
                 print('"{}" [shape=box fillcolor=gray style=filled]'.format(
                       node.name))