Skip to content

Commit 9982f49

Browse files
committed
fix error
1 parent d38be93 commit 9982f49

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

nipype/interfaces/dipy/preprocess.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,10 @@ def _run_interface(self, runtime):
150150
if isdefined(self.inputs.block_radius):
151151
settings['block_radius'] = self.inputs.block_radius
152152

153+
snr = None
154+
if isdefined(self.inputs.snr):
155+
snr = self.inputs.snr
156+
153157
signal_mask = None
154158
if isdefined(self.inputs.signal_mask):
155159
signal_mask = nb.load(self.inputs.signal_mask).get_data()
@@ -158,6 +162,7 @@ def _run_interface(self, runtime):
158162
noise_mask = nb.load(self.inputs.noise_mask).get_data()
159163

160164
_, s = nlmeans_proxy(self.inputs.in_file, settings,
165+
snr=snr,
161166
smask=signal_mask,
162167
nmask=noise_mask,
163168
out_file=out_file)
@@ -215,6 +220,7 @@ def resample_proxy(in_file, order=3, new_zooms=None, out_file=None):
215220

216221

217222
def nlmeans_proxy(in_file, settings,
223+
snr=None,
218224
smask=None,
219225
nmask=None,
220226
out_file=None):
@@ -281,12 +287,13 @@ def nlmeans_proxy(in_file, settings,
281287
nmask = binary_erosion(nmask, iterations=1).astype(np.uint8)
282288

283289
den = np.zeros_like(data)
284-
snr = []
285290

286291
est_snr = True
287-
if isdefined(self.inputs.snr):
292+
if snr is not None:
288293
snr = [self.inputs.snr] * data.shape[-1]
289294
est_snr = False
295+
else:
296+
snr = []
290297

291298
for i in range(data.shape[-1]):
292299
d = data[..., i]

0 commit comments

Comments
 (0)