fix some py3-isms => py2 compat

sourmash-bio · Nov 27, 2016 · 422943f · 422943f
1 parent f053ae3
commit 422943f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/sourmash_lib/sbt.py b/sourmash_lib/sbt.py
@@ -1,5 +1,4 @@
 #!/usr/bin/env python
-
 """
 A trial implementation of sequence bloom trees, Solomon & Kingsford, 2015.
 
@@ -119,7 +118,7 @@ def add_node(self, node):
             n = Node(self.factory, name="internal." + str(p.pos))
             self.nodes[p.pos] = n
 
-            c1, c2, *remainder = self.children(p.pos)
+            c1, c2 = self.children(p.pos)[:2]
 
             self.nodes[c1.pos] = p.node
             self.nodes[c2.pos] = node

diff --git a/sourmash_lib/sbtmh.py b/sourmash_lib/sbtmh.py
@@ -1,3 +1,4 @@
+from __future__ import print_function
 from glob import glob
 import os
 

diff --git a/sourmash_lib/signature.py b/sourmash_lib/signature.py
@@ -142,7 +142,7 @@ def load_signatures(data, select_ksize=None, ignore_md5sum=False):
     # at this point, whatever 'data' is, it should be loadable!
 
     # record header
-    x = yaml.safe_load_all(data)
+    x = yaml.load_all(data)
     siglist = []
     for d in x:      # allow empty records <-> concatenation of signatures
         if not d: