Skip to content

Commit 5f87526

Browse files
committed
revert changes to multiproc - to another PR
1 parent 5cb476e commit 5f87526

File tree

1 file changed

+6
-15
lines changed

1 file changed

+6
-15
lines changed

nipype/pipeline/plugins/multiproc.py

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,10 @@
1212
from multiprocessing import Process, Pool, cpu_count, pool
1313
from traceback import format_exception
1414
import sys
15-
from textwrap import indent
16-
from logging import INFO
1715

1816
from copy import deepcopy
1917
import numpy as np
18+
2019
from ... import logging
2120
from ...utils.profiler import get_system_total_memory_gb
2221
from ..engine import MapNode
@@ -127,7 +126,7 @@ def __init__(self, plugin_args=None):
127126
self.raise_insufficient = self.plugin_args.get('raise_insufficient', True)
128127

129128
# Instantiate different thread pools for non-daemon processes
130-
logger.debug('[MultiProc] Starting in "%sdaemon" mode (n_procs=%d, mem_gb=%0.2f)',
129+
logger.debug('MultiProcPlugin starting in "%sdaemon" mode (n_procs=%d, mem_gb=%0.2f)',
131130
'non' * int(non_daemon), self.processors, self.memory_gb)
132131

133132
NipypePool = NonDaemonPool if non_daemon else Pool
@@ -159,7 +158,7 @@ def _submit_job(self, node, updatehash=False):
159158
run_node, (node, updatehash, self._taskid),
160159
callback=self._async_callback)
161160

162-
logger.debug('[MultiProc] Submitted task %s (taskid=%d).',
161+
logger.debug('MultiProc submitted task %s (taskid=%d).',
163162
node.fullname, self._taskid)
164163
return self._taskid
165164

@@ -215,17 +214,9 @@ def _send_procs_to_workers(self, updatehash=False, graph=None):
215214
stats = (len(self.pending_tasks), len(jobids), free_memory_gb,
216215
self.memory_gb, free_processors, self.processors)
217216
if self._stats != stats:
218-
tasks_list_msg = ''
219-
if logger.level <= INFO:
220-
running_tasks = [' * %s' % self.procs[jobid].fullname
221-
for _, jobid in self.pending_tasks]
222-
if running_tasks:
223-
tasks_list_msg = '\nCurrently running:\n'
224-
tasks_list_msg += '\n'.join(running_tasks)
225-
tasks_list_msg = indent(tasks_list_msg, ' ' * 21)
226-
logger.info('[MultiProc] Running %d tasks, and %d jobs ready. Free '
227-
'memory (GB): %0.2f/%0.2f, Free processors: %d/%d.%s',
228-
*stats, tasks_list_msg)
217+
logger.info('Currently running %d tasks, and %d jobs ready. Free '
218+
'memory (GB): %0.2f/%0.2f, Free processors: %d/%d',
219+
*stats)
229220
self._stats = stats
230221

231222
if free_memory_gb < 0.01 or free_processors == 0:

0 commit comments

Comments
 (0)