Skip to content

Commit a34d50f

Browse files
authored
Merge pull request #2010 from oesteban/enh/AntsAffineInitializerThreads
[ENH] Enable num_threads for antsAffineInitializer
2 parents 4358fe5 + f82d4c8 commit a34d50f

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

nipype/interfaces/ants/tests/test_auto_AffineInitializer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@ def test_AffineInitializer_inputs():
2828
mandatory=True,
2929
position=2,
3030
),
31+
num_threads=dict(nohash=True,
32+
usedefault=True,
33+
),
3134
out_file=dict(argstr='%s',
3235
position=3,
3336
usedefault=True,

nipype/interfaces/ants/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def _list_outputs(self):
181181
return outputs
182182

183183

184-
class AffineInitializerInputSpec(CommandLineInputSpec):
184+
class AffineInitializerInputSpec(ANTSCommandInputSpec):
185185
dimension = traits.Enum(3, 2, usedefault=True, position=0, argstr='%s',
186186
desc='dimension')
187187
fixed_image = File(exists=True, mandatory=True, position=1, argstr='%s',
@@ -207,7 +207,7 @@ class AffineInitializerOutputSpec(TraitedSpec):
207207
out_file = File(desc='output transform file')
208208

209209

210-
class AffineInitializer(CommandLine):
210+
class AffineInitializer(ANTSCommand):
211211
"""
212212
Initialize an affine transform (as in antsBrainExtraction.sh)
213213

0 commit comments

Comments
 (0)