From 5dd1cf01d1a6757ec4aac0c0063cb1202ba1fb63 Mon Sep 17 00:00:00 2001
From: Chris Maffeo <cmaffeo2@illinois.edu>
Date: Tue, 4 Sep 2018 14:04:44 -0500
Subject: [PATCH] Fixed bug in how beads are assigned a type when clustering
 fails

---
 dnarbd/segmentmodel.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/dnarbd/segmentmodel.py b/dnarbd/segmentmodel.py
index 784d792..cf82cd3 100644
--- a/dnarbd/segmentmodel.py
+++ b/dnarbd/segmentmodel.py
@@ -1707,8 +1707,8 @@ class SegmentModel(ArbdModel):
             clusters = hcluster.fclusterdata(data, float(max_basepairs_per_bead)/500, criterion="distance")
             cluster_size = [np.mean(data[clusters == i]) for i in np.unique(clusters)]
         except:
-            clusters = data
-            cluster_size = np.arange(len(data))+1
+            clusters = np.arange(len(data))+1
+            cluster_size = data.flatten()
         for b,c in zip(beads,clusters):
             _assign_bead_type(b, cluster_size[c-1], decimals=order)
 
@@ -1719,8 +1719,8 @@ class SegmentModel(ArbdModel):
             clusters = hcluster.fclusterdata(data, float(max_nucleotides_per_bead)/500, criterion="distance")
             cluster_size = [np.mean(data[clusters == i]) for i in np.unique(clusters)]
         except:
-            clusters = data
-            cluster_size = np.arange(len(data))+1
+            clusters = np.arange(len(data))+1
+            cluster_size = data.flatten()
         for b,c in zip(beads,clusters):
             _assign_bead_type(b, cluster_size[c-1], decimals=order)
 
-- 
GitLab