Single reference compatibitlity

betteridiot · Mar 19, 2019 · 72addad · 72addad
1 parent 50f1877
commit 72addad
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/bamnostic/core.py b/bamnostic/core.py
@@ -273,7 +273,11 @@ def _unpack_data(self):
         self.read_name = unpack('<{}s'.format(self._l_read_name), self._range_popper(self._l_read_name)).decode()[:-1]
 
         self.tid = self.reference_id = self.refID
-        self.reference_name = self._io._header.refs[self.refID][0]
+        try:
+            self.reference_name = self._io._header.refs[self.refID][0]
+        except KeyError:
+            if self.refID == -1 and len(self._io._header.refs) == 1:
+                self.reference_name = self._io._header.refs[0][0]
 
     def _cigar_builder(self):
         """Just unpacks the cigar data to be processed later. Ensures the cursor