add psij subtype to WORKERS

Author: adi611

pydra/engine/workers.py

@@ -893,14 +893,15 @@ def close(self):
 
 
 class PsijWorker(Worker):
-    def __init__(self, **kwargs):
+    def __init__(self, subtype=None, **kwargs):
         """Initialize worker."""
         try:
             import psij
         except ImportError:
             logger.critical("Please install psij.")
             raise
         logger.debug("Initialize PsijWorker")
+        self.subtype = subtype
 
     def run_el(self, interface, rerun=False, **kwargs):
         """Run a task."""
@@ -926,7 +927,7 @@ async def exec_psij(self, runnable, rerun=False):
         import os
 
         self.psij = psij
-        jex = psij.JobExecutor.get_instance("slurm")
+        jex = psij.JobExecutor.get_instance(self.subtype)
         absolute_path = os.path.dirname(__file__)
 
         if isinstance(runnable, TaskBase):
@@ -976,5 +977,5 @@ def close(self):
     "slurm": SlurmWorker,
     "dask": DaskWorker,
     "sge": SGEWorker,
-    "psij": PsijWorker,
+    **{"psij-" + subtype: lambda subtype=subtype: PsijWorker(subtype=subtype) for subtype in ["local", "slurm"]},
 }