BUG: Fix signature for _run_task

nipype · Mar 22, 2024 · 5504be7 · 5504be7
1 parent 80c7b02
commit 5504be7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pydra/engine/core.py b/pydra/engine/core.py
@@ -380,7 +380,7 @@ def can_resume(self):
         return self._can_resume
 
     @abc.abstractmethod
-    def _run_task(self):
+    def _run_task(self, environment=None):
         pass
 
     @property
@@ -1329,7 +1329,7 @@ async def _run(self, submitter=None, rerun=False, **kwargs):
         self._check_for_hash_changes()
         return result
 
-    async def _run_task(self, submitter, rerun=False):
+    async def _run_task(self, submitter, rerun=False, environment=None):
         if not submitter:
             raise Exception("Submitter should already be set.")
         for nd in self.graph.nodes: