replace list stringd by existing enum

Author: paulnoirel

libs/labelbox/src/labelbox/schema/workflow/workflow.py

@@ -61,7 +61,7 @@
 def _validate_definition_id(
     definition_id_str: str, node_id: str
 ) -> WorkflowDefinitionId:
-    """Validate and normalize a workflow definition ID with consistent error handling.
+    """Validate and normalize a workflow definition ID.
 
     Args:
         definition_id_str: The definition ID string to validate
@@ -71,26 +71,10 @@ def _validate_definition_id(
         WorkflowDefinitionId: Validated definition ID or fallback
     """
     try:
-        if definition_id_str in (
-            "initial_labeling_task",
-            "initial_rework_task",
-            "review_task",
-            "send_to_rework",
-            "logic",
-            "done",
-            "custom_rework_task",
-            "auto_qa",
-        ):
-            return WorkflowDefinitionId(definition_id_str)
-        else:
-            logger.warning(
-                f"Unknown WorkflowDefinitionId '{definition_id_str}' for node {node_id}. "
-                f"Using InitialLabelingTask as fallback."
-            )
-            return WorkflowDefinitionId.InitialLabelingTask
-    except ValueError:
+        return WorkflowDefinitionId(definition_id_str)
+    except ValueError as e:
         logger.warning(
-            f"Invalid WorkflowDefinitionId '{definition_id_str}' for node {node_id}. "
+            f"Invalid WorkflowDefinitionId '{definition_id_str}' for node {node_id}: {e}. "
             f"Using InitialLabelingTask as fallback."
         )
         return WorkflowDefinitionId.InitialLabelingTask