Fix: Set ValidationWorkflow as a WrapperTask (#54)

Author: adrien-berchet

data_validation_framework/task.py

@@ -707,7 +707,7 @@ def _process(self, df, *args, **kwargs):
         return df
 
 
-class ValidationWorkflow(SetValidationTask):
+class ValidationWorkflow(SetValidationTask, luigi.WrapperTask):
     """Class to define and process a validation workflow."""
 
     report_path = OptionalStrParameter(

tests/test_task.py

@@ -2520,7 +2520,10 @@ class TestTask1(task.SkippableMixin(), luigi.Task):
         ):
             TestTask1()
 
-        class TestTask2(task.SkippableMixin(), task.ValidationWorkflow):
+        class TestTask2(
+            task.SkippableMixin(),
+            task.ValidationWorkflow,
+        ):  # pylint: disable=too-many-ancestors
             pass
 
         with pytest.raises(