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
index 02d6c1210..8499e6f63 100644
--- a/src/dioptra/restapi/v1/workflows/lib/export_task_engine_yaml.py
+++ b/src/dioptra/restapi/v1/workflows/lib/export_task_engine_yaml.py
@@ -88,6 +88,16 @@ def build_task_engine_dict_for_validation(
     parameters: dict[str, Any],
     task_graph: str,
 ) -> dict[str, Any]:
+    """Build a dictionary representation of a task engine YAML file.
+
+    Args:
+        plugins: The entrypoint's plugin files.
+        parameters: The entrypoint parameteres.
+        task_graph: The task graph of the entrypoint.
+
+    Returns:
+        The task engine dictionary.
+    """
     tasks: dict[str, Any] = {}
     parameter_types: dict[str, Any] = {}
     for plugin in plugins: