Use FlyteValidationException instead of AssertionError

Signed-off-by: Fabio Grätz <[email protected]>

flytekit/core/workflow.py

@@ -756,7 +756,7 @@ def compile(self, **kwargs):
                 )
                 bindings.append(b)
             except Exception as e:
-                raise AssertionError(f"Failed to bind output {output_names[0]} for function {self.name}: {e}") from e
+                raise FlyteValidationException(f"Failed to bind output {output_names[0]} for function {self.name}: {e}") from e
         elif len(output_names) > 1:
             if not isinstance(workflow_outputs, tuple):
                 raise AssertionError("The Workflow specification indicates multiple return values, received only one")
@@ -776,7 +776,7 @@ def compile(self, **kwargs):
                     )
                     bindings.append(b)
                 except Exception as e:
-                    raise AssertionError(f"Failed to bind output {out} for function {self.name}: {e}") from e
+                    raise FlyteValidationException(f"Failed to bind output {out} for function {self.name}: {e}") from e
 
         # Save all the things necessary to create an WorkflowTemplate, except for the missing project and domain
         self._nodes = all_nodes

tests/flytekit/unit/core/test_type_hints.py

@@ -34,6 +34,7 @@
 from flytekit.core.testing import patch, task_mock
 from flytekit.core.type_engine import RestrictedTypeError, SimpleTransformer, TypeEngine
 from flytekit.core.workflow import workflow
+from flytekit.exceptions.user import FlyteValidationException
 from flytekit.models import literals as _literal_models
 from flytekit.models.core import types as _core_types
 from flytekit.models.interface import Parameter
@@ -138,7 +139,7 @@ def test_missing_output():
     def wf() -> str:
         return None  # type: ignore
 
-    with pytest.raises(AssertionError, match="Failed to bind output"):
+    with pytest.raises(FlyteValidationException, match="Failed to bind output"):
         wf.compile()
 
 

tests/flytekit/unit/experimental/test_eager_workflows.py

@@ -9,6 +9,7 @@
 from hypothesis import given, settings
 
 from flytekit import dynamic, task, workflow
+from flytekit.exceptions.user import FlyteValidationException
 from flytekit.experimental import EagerException, eager
 from flytekit.types.directory import FlyteDirectory
 from flytekit.types.file import FlyteFile
@@ -212,7 +213,7 @@ async def eager_wf(x: int) -> int:
         out = await local_wf(x=x)
         return await double(x=out)
 
-    with pytest.raises(AssertionError):
+    with pytest.raises(FlyteValidationException):
         asyncio.run(eager_wf(x=x_input))