Skip to content

Commit 2bc7a64

Browse files
authored
Merge pull request #142 from oesteban/fix/n-procs
FIX: Honor ``--nprocs`` flag
2 parents 44fec84 + dda3d4e commit 2bc7a64

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

dmriprep/cli/parser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -439,7 +439,7 @@ def parse_args(args=None, namespace=None):
439439
config.nipype.plugin = _plugin
440440
config.nipype.plugin_args = plugin_settings.get("plugin_args", {})
441441
config.nipype.nprocs = config.nipype.plugin_args.get(
442-
"nprocs", config.nipype.nprocs
442+
"n_procs", config.nipype.nprocs
443443
)
444444

445445
# Resource management options

dmriprep/config/__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -279,12 +279,17 @@ class nipype(_Config):
279279
@classmethod
280280
def get_plugin(cls):
281281
"""Format a dictionary for Nipype consumption."""
282+
nprocs = int(cls.nprocs)
283+
if nprocs == 1:
284+
cls.plugin = "Linear"
285+
return {"plugin": "Linear"}
286+
282287
out = {
283288
"plugin": cls.plugin,
284289
"plugin_args": cls.plugin_args,
285290
}
286291
if cls.plugin in ("MultiProc", "LegacyMultiProc"):
287-
out["plugin_args"]["nprocs"] = int(cls.nprocs)
292+
out["plugin_args"]["n_procs"] = int(cls.nprocs)
288293
if cls.memory_gb:
289294
out["plugin_args"]["memory_gb"] = float(cls.memory_gb)
290295
return out

0 commit comments

Comments
 (0)