Use features from luigi >= 3.1 instead of luigi-tools (#26)

Author: adrien-berchet

data_validation_framework/task.py

@@ -11,13 +11,12 @@
 import pandas as pd
 from luigi.parameter import ChoiceParameter
 from luigi.parameter import DictParameter
+from luigi.parameter import OptionalIntParameter
+from luigi.parameter import OptionalPathParameter
+from luigi.parameter import OptionalStrParameter
 from luigi.task import flatten as task_flatten
 from luigi_tools.parameter import BoolParameter
 from luigi_tools.parameter import OptionalBoolParameter
-from luigi_tools.parameter import OptionalIntParameter
-from luigi_tools.parameter import OptionalPathParameter
-from luigi_tools.parameter import OptionalStrParameter
-from luigi_tools.task import CheckUnconsumedParamsMixin
 from luigi_tools.task import LogTargetMixin
 from luigi_tools.task import RerunMixin
 from numpy import VisibleDeprecationWarning
@@ -122,7 +121,6 @@ def __init__(self, col_mapping, **kwargs):
 
 
 class BaseValidationTask(
-    CheckUnconsumedParamsMixin,
     LogTargetMixin,
     RerunMixin,
     TagResultOutputMixin,

setup.py

@@ -7,7 +7,7 @@
     README = f.read()
 
 reqs = [
-    "luigi",
+    "luigi>=3.1",
     "luigi-tools>=0.0.18",
     "pandas",
     "rst2pdf",

tests/test_task.py

@@ -19,7 +19,7 @@
 import pause
 import pytest
 from diff_pdf_visually import pdfdiff
-from luigi_tools.parameter import OptionalStrParameter
+from luigi.parameter import OptionalStrParameter
 
 from data_validation_framework import report
 from data_validation_framework import result