add working remove mers from file

1 files changed, 6 insertions, 3 deletions

diff --git a/src/remove_mers_from_file.py b/src/remove_mers_from_file.py
index bef3ea3..3576e35 100755
--- a/src/remove_mers_from_file.py
+++ b/src/remove_mers_from_file.py
@@ -8,9 +8,12 @@ if __name__ == "__main__":
 	
 	mers_to_delete = set()
 
-	for mer in open(sys.argv[1], 'r'):
-		mer = mer.strip().split()[0]
-		mers_to_delete.add(mer)
+	for mer in open(sys.argv[1], 'rU'):
+		sp = mer.strip().split()
+		if len(sp) == 2:
+			mer = sp[0]
+			mers_to_delete.add(mer)
+
 
 	for line in sys.stdin:
 		if line.split()[0] not in mers_to_delete: