additional task_fields->get_fields changes

Author: tclose

pydra/engine/node.py

@@ -114,13 +114,13 @@ def state_values(self) -> dict[str, ty.Any]:
 
     @property
     def lzout(self) -> OutputType:
-        from pydra.utils.general import task_fields
+        from pydra.utils.general import get_fields
 
         """The output task of the node populated with lazy fields"""
         if self._lzout is not None:
             return self._lzout
         lazy_fields = {}
-        for field in task_fields(self.inputs.Outputs):
+        for field in get_fields(self.inputs.Outputs):
             lazy_fields[field.name] = lazy.LazyOutField(
                 node=self,
                 field=field.name,

pydra/engine/tests/utils.py

@@ -9,7 +9,7 @@
 import subprocess as sp
 import pytest
 from fileformats.generic import File
-from pydra.utils.general import task_fields
+from pydra.utils.general import get_fields
 from pydra.engine.submitter import Submitter
 from pydra.compose import workflow, python, shell
 
@@ -44,7 +44,7 @@ def num_python_cache_roots(cache_path: Path) -> int:
 
 
 def get_output_names(task):
-    return sorted(f.name for f in task_fields(task.Outputs))
+    return sorted(f.name for f in get_fields(task.Outputs))
 
 
 def run_no_submitter(