From a30592a22aa693723b1d0efe74f7baceba7abbe9 Mon Sep 17 00:00:00 2001 From: "Gregory R. Lee" Date: Sat, 23 Jan 2016 11:23:21 -0500 Subject: [PATCH] MAINT: explicitly specify axis argument --- skimage/restoration/_denoise.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skimage/restoration/_denoise.py b/skimage/restoration/_denoise.py index 9e06fbd8..acd25cb0 100644 --- a/skimage/restoration/_denoise.py +++ b/skimage/restoration/_denoise.py @@ -178,7 +178,7 @@ def _denoise_tv_chambolle_nd(im, weight=0.1, eps=2.e-4, n_iter_max=200): g[slices_g] = np.diff(out, axis=ax) slices_g[ax+1] = slice(None) - norm = np.sqrt((g ** 2).sum(0))[np.newaxis, ...] + norm = np.sqrt((g ** 2).sum(axis=0))[np.newaxis, ...] E += weight * norm.sum() tau = 1. / (2.*ndim) norm *= tau / weight