fix: ensure task input and output parameters are sorted correctly

usnistgov · Sep 12, 2024 · f4f4b99 · f4f4b99
1 parent 24f00b2
commit f4f4b99
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/dioptra/restapi/v1/workflows/lib/export_task_engine_yaml.py b/src/dioptra/restapi/v1/workflows/lib/export_task_engine_yaml.py
@@ -173,8 +173,12 @@ def extract_tasks(
         plugin_file = entry_point_plugin_file.plugin_file
 
         for task in plugin_file.tasks:
-            input_parameters = task.input_parameters
-            output_parameters = task.output_parameters
+            input_parameters = sorted(
+                task.input_parameters, key=lambda x: x.parameter_number
+            )
+            output_parameters = sorted(
+                task.output_parameters, key=lambda x: x.parameter_number
+            )
 
             tasks[task.plugin_task_name] = {
                 "plugin": _build_plugin_field(plugin, plugin_file, task),