StringIO,BytesIO compatibility fixed for py26/27 py3

VDBWRAIR · Aug 12, 2015 · d5b0e74 · d5b0e74
1 parent 21b1a2a
commit d5b0e74
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/bio_pieces/sns_matrix.py b/bio_pieces/sns_matrix.py
@@ -1,12 +1,15 @@
 import pandas as pd
 import numpy as np
-from StringIO import StringIO
+try:
+    from io import StringIO as BytesIO
+except ImportError:
+    from io import BytesIO
 
 
 # Simple Nucleotide Substitution matrix
 # Copy/pasted table from 
 # http://www.jalview.org/help/html/calculations/scorematrices.html#simplenucleotide
-sns = StringIO('''
+sns = BytesIO(u'''
 ,A,C,G,I,N,R,T,U,X,Y
 A,10,-8,-8,1,1,1,-8,-8,1,-8
 C,-8,10,-8,1,1,-8,-8,-8,1,1