Skip to content

Commit 5b04d99

Browse files
fix: incorrect arguments to np.pad
1 parent d00566e commit 5b04d99

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

kimimaro/utility.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -194,12 +194,12 @@ def moving_average(a:np.ndarray, n:int) -> np.ndarray:
194194
raise ValueError(f"Window size ({n}), must be >= 1.")
195195
elif n == 1:
196196
return a
197-
mirror = (len(a) - (len(a) - n + 1)) / 2
197+
mirror = (n - 1) / 2
198198
extra = 0
199199
if mirror != int(mirror):
200200
extra = 1
201201
mirror = int(mirror)
202-
a = np.pad(a, [[mirror+extra, extra],[0,0]], mode="symmetric")
202+
a = np.pad(a, [[mirror, mirror+extra],[0,0]], mode="symmetric")
203203
ret = np.cumsum(a, dtype=float, axis=0)
204204
ret[n:] = ret[n:] - ret[:-n]
205205
return ret[n - 1:] / n

0 commit comments

Comments
 (0)