fixed structure_designer

freude · Jul 5, 2020 · 1f9b198 · 1f9b198
1 parent fbc50db
commit 1f9b198
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nanonet/tb/structure_designer.py b/nanonet/tb/structure_designer.py
@@ -103,9 +103,9 @@ def _sort(self, labels, coords):
                                  mat=h_matrix)
 
         self.reorder = indices
-        if isinstance(self.left_lead, list) or isinstance(self.left_lead, np.ndarray) :
+        if isinstance(self.left_lead, list) or isinstance(self.left_lead, np.ndarray) and len(self.left_lead) > 0:
             self.left_lead = np.squeeze(np.concatenate([np.where(indices == item) for item in self.left_lead]))
-        if isinstance(self.right_lead, list) or isinstance(self.right_lead, np.ndarray) :
+        if isinstance(self.right_lead, list) or isinstance(self.right_lead, np.ndarray) and len(self.left_lead) > 0:
             self.right_lead = np.squeeze(np.concatenate([np.where(indices == item) for item in self.right_lead]))
         coords = coords[indices]
         labels = [labels[i] for i in indices]