Skip to content

Commit 1766f00

Browse files
committed
Add padding as well
1 parent 00419b4 commit 1766f00

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/diffpy/morph/refine.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,12 +91,17 @@ def _residual(self, pvals):
9191
else:
9292
# Padding
9393
if len(rvec) < self.res_length:
94+
diff_length = self.res_length - len(rvec)
9495
rvec = list(rvec)
95-
rvec.extend([0] * (self.res_length - len(rvec)))
96+
rvec.extend([0] * diff_length)
9697
rvec = array(rvec)
9798
# Removal
9899
elif len(rvec) > self.res_length:
99-
pass
100+
diff_length = len(rvec) - self.res_length
101+
rvec = list(rvec)
102+
for _ in range(diff_length):
103+
rvec.pop(max(rvec))
104+
rvec = array(rvec)
100105

101106
return rvec
102107

0 commit comments

Comments
 (0)