From f6b776773fbb41cfcdec992f994b1da4f20bff20 Mon Sep 17 00:00:00 2001
From: Chris Maffeo <cmaffeo2@illinois.edu>
Date: Tue, 18 Feb 2020 12:35:07 -0600
Subject: [PATCH] Improved performance for segmentmodel._getParent()
---
mrdna/segmentmodel.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/mrdna/segmentmodel.py b/mrdna/segmentmodel.py
index 74c7782..5ccbe93 100644
--- a/mrdna/segmentmodel.py
+++ b/mrdna/segmentmodel.py
@@ -1690,8 +1690,7 @@ class SegmentModel(ArbdModel):
def _getParent(self, *beads ):
- if np.all( [b1.parent == b2.parent
- for b1,b2 in zip(beads[:-1],beads[1:])] ):
+ if len(set([b.parent for b in beads])) == 1:
return beads[0].parent
else:
return self
--
GitLab