updated generate script to latest version of pydra helper function locations

Author: tclose

generate.py

@@ -18,7 +18,7 @@
 from pydra.compose import shell
 from pydra.compose.base import NO_DEFAULT
 from pydra.utils.typing import MultiInputObj
-from pydra.utils.general import task_fields, add_exc_note
+from pydra.utils.general import get_fields, add_exc_note
 
 
 logger = logging.getLogger("pydra-auto-gen")
@@ -314,8 +314,8 @@ def auto_gen_test(cmd_name: str, output_dir: Path, log_errors: bool, pkg_version
     module = import_module(f"pydra.tasks.mrtrix3.{pkg_version}.{cmd_name}")
     definition_klass = getattr(module, pascal_case_task_name(cmd_name))
 
-    input_fields = task_fields(definition_klass)
-    output_fields = task_fields(definition_klass.Outputs)
+    input_fields = get_fields(definition_klass)
+    output_fields = get_fields(definition_klass.Outputs)
     output_fields_dict = {f.name: f for f in output_fields}
 
     code_str = f"""# Auto-generated test for {cmd_name}

pyproject.toml

@@ -84,3 +84,6 @@ strict = true
 namespace_packages = true
 explicit_package_bases = true
 exclude = ["tests", "scripts", "docs", "build", "dist"]
+
+[tool.pyright]
+reportAssignmentType = false