code refactory

Author: mauriliogenovese

nipype/pipeline/engine/nodes.py

@@ -823,7 +823,8 @@ def update(self, **opts):
 
     def is_gpu_node(self):
         return bool(getattr(self.inputs, 'use_cuda', False)) or bool(
-            getattr(self.inputs, 'use_gpu', False))
+            getattr(self.inputs, 'use_gpu', False)
+        )
 
 
 class JoinNode(Node):

nipype/pipeline/plugins/multiproc.py

@@ -141,7 +141,9 @@ def __init__(self, plugin_args=None):
         if self.n_gpu_procs > self.n_gpus_visible:
             logger.info(
                 'Total number of GPUs proc requested (%d) exceeds the available number of GPUs (%d) on the system. Using requested GPU slots at your own risk!',
-                self.n_gpu_procs, self.n_gpus_visible)
+                self.n_gpu_procs,
+                self.n_gpus_visible,
+            )
 
         # Instantiate different thread pools for non-daemon processes
         logger.debug(
@@ -394,7 +396,7 @@ def _send_procs_to_workers(self, updatehash=False, graph=None):
                 free_memory_gb += next_job_gb
                 free_processors += next_job_th
                 if is_gpu_node:
-                    free_gpu_slots -= next_job_gpu_th
+                    free_gpu_slots += next_job_gpu_th
                 # Display stats next loop
                 self._stats = None
 
@@ -421,5 +423,4 @@ def _sort_jobs(self, jobids, scheduler="tsort"):
                 jobids,
                 key=lambda item: (self.procs[item].mem_gb, self.procs[item].n_procs),
             )
-        return jobids
-
+        return jobids

nipype/pipeline/plugins/tools.py

@@ -179,11 +179,12 @@ def create_pyscript(node, updatehash=False, store_exception=True):
         fp.writelines(cmdstr)
     return pyscript
 
+
 def gpu_count():
     n_gpus = 1
     try:
         import GPUtil
     except ImportError:
         return 1
     else:
-        return len(GPUtil.getGPUs())
+        return len(GPUtil.getGPUs())