Skip to content

Commit 81b738b

Browse files
committed
moving print to warning and editing docstring
1 parent 1ed121d commit 81b738b

File tree

3 files changed

+12
-17
lines changed

3 files changed

+12
-17
lines changed

nipype/interfaces/niftyreg/base.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -23,16 +23,12 @@
2323
from distutils.version import StrictVersion
2424
import os
2525
import subprocess
26-
import warnings
26+
from warnings import warn
2727

2828
from ..base import CommandLine, isdefined, traits
2929
from ...utils.filemanip import split_filename
3030

3131

32-
warn = warnings.warn
33-
warnings.filterwarnings('always', category=UserWarning)
34-
35-
3632
def get_custom_path(command):
3733
try:
3834
specific_dir = os.environ['NIFTYREGDIR']
@@ -77,13 +73,13 @@ def __init__(self, required_version=None, **inputs):
7773
if _version:
7874
_version = _version.decode("utf-8")
7975
if StrictVersion(_version) < StrictVersion(self._min_version):
80-
warn = 'A later version of Niftyreg is required (%s < %s)'
81-
print(warn % (_version, self._min_version))
76+
msg = 'A later version of Niftyreg is required (%s < %s)'
77+
warn(msg % (_version, self._min_version))
8278
if required_version is not None:
8379
if StrictVersion(_version) != StrictVersion(required_version):
84-
warn = 'The version of NiftyReg differs from the required'
85-
warn += '(%s != %s)'
86-
print(warn % (_version, required_version))
80+
msg = 'The version of NiftyReg differs from the required'
81+
msg += '(%s != %s)'
82+
warn(msg % (_version, self.required_version))
8783

8884
def check_version(self):
8985
_version = self.get_version()

nipype/interfaces/niftyreg/reg.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ class RegAladin(NiftyRegCommand):
157157
>>> node.inputs.omp_core_val = 4
158158
>>> node.cmdline # doctest: +ELLIPSIS +ALLOW_UNICODE
159159
'reg_aladin -aff .../im2_aff.txt -flo im2.nii -omp 4 -ref im1.nii \
160-
-rmask mask.nii -res .../im2_res.nii.gz'
160+
-res .../im2_res.nii.gz -rmask mask.nii'
161161
162162
"""
163163
_cmd = get_custom_path('reg_aladin')
@@ -386,9 +386,8 @@ class RegF3D(NiftyRegCommand):
386386
>>> node.inputs.rmask_file = 'mask.nii'
387387
>>> node.inputs.omp_core_val = 4
388388
>>> node.cmdline # doctest: +ELLIPSIS +ALLOW_UNICODE
389-
'reg_f3d -be 0.100000 -cpp .../im2_cpp.nii.gz -flo im2.nii -le \
390-
0.100000 -omp 4 -ref im1.nii -res .../flo_res.nii.gz -rmask mask.nii \
391-
-vel'
389+
'reg_f3d -cpp .../im2_cpp.nii.gz -flo im2.nii -omp 4 -ref im1.nii \
390+
-res .../flo_res.nii.gz -rmask mask.nii'
392391
393392
"""
394393
_cmd = get_custom_path('reg_f3d')

nipype/interfaces/niftyreg/regutils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ class RegResample(NiftyRegCommand):
110110
>>> node.inputs.inter_val = 'LIN'
111111
>>> node.inputs.omp_core_val = 4
112112
>>> node.cmdline # doctest: +ELLIPSIS +ALLOW_UNICODE
113-
'reg_resample -flo im2.nii.gz -inter 1 -omp 4 -ref im1.nii -trans \
113+
'reg_resample -flo im2.nii -inter 1 -omp 4 -ref im1.nii -trans \
114114
warpfield.nii -res .../im2_res.nii.gz'
115115
116116
"""
@@ -409,8 +409,8 @@ class RegAverage(NiftyRegCommand):
409409
>>> two_file = 'im2.nii'
410410
>>> three_file = 'im3.nii'
411411
>>> node.inputs.avg_files = [one_file, two_file, three_file]
412-
>>> node.cmdline # doctest: +ALLOW_UNICODE
413-
'reg_average --cmd_file reg_average_cmd'
412+
>>> node.cmdline # doctest: +ELLIPSIS +ALLOW_UNICODE
413+
'reg_average --cmd_file .../reg_average_cmd'
414414
415415
"""
416416
_cmd = get_custom_path('reg_average')

0 commit comments

Comments
 (0)