Modify BurrowsWheelerLibrary.py

AlphaGenes · May 20, 2024 · 9fa1964 · 9fa1964
1 parent 14a2cd8
commit 9fa1964
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/alphaimpute2/Imputation/BurrowsWheelerLibrary.py b/src/alphaimpute2/Imputation/BurrowsWheelerLibrary.py
@@ -59,6 +59,10 @@ def removeMissingValues(self):
     def setup_library(self, loci=None, create_reverse_library=False, create_a=False):
         self.removeMissingValues()
         self.haplotypes = np.array(self.haplotypes)
+
+        if self.haplotypes.size == 0:
+            return
+
         if loci is not None:
             bw_loci = np.array(loci, dtype=np.int64)
             sub_library = self.haplotypes[:, bw_loci]