diff --git a/gokart/parameter.py b/gokart/parameter.py index c83c4290..7b1d3062 100644 --- a/gokart/parameter.py +++ b/gokart/parameter.py @@ -120,7 +120,7 @@ def _warn_on_wrong_param_type(self, param_name, param_value): class ExplicitBoolParameter(luigi.BoolParameter): def __init__(self, *args, **kwargs): - luigi.Parameter.__init__(self, *args, **kwargs) + super(luigi.BoolParameter, self).__init__(*args, **kwargs) def _parser_kwargs(self, *args, **kwargs): # type: ignore return luigi.Parameter._parser_kwargs(*args, *kwargs) diff --git a/gokart/task.py b/gokart/task.py index c6afc517..0ee838dd 100644 --- a/gokart/task.py +++ b/gokart/task.py @@ -402,7 +402,7 @@ def _to_str_params(task): return task.to_str_params(only_significant=True) - dependencies = [_to_str_params(task) for task in flatten(self.requires())] + dependencies: list[Any] = [_to_str_params(task) for task in flatten(self.requires())] dependencies = [d for d in dependencies if d is not None] dependencies.append(self.to_str_params(only_significant=True)) dependencies.append(self.__class__.__name__) @@ -445,7 +445,7 @@ def get_info(self, only_significant=False): for param_name, param_value in self.param_kwargs.items(): if (not only_significant) or params[param_name].significant: if isinstance(params[param_name], gokart.TaskInstanceParameter): - params_str[param_name] = type(param_value).__name__ + '-' + param_value.make_unique_id() + params_str[param_name] = type(param_value).__name__ + '-' + cast(TaskOnKart[Any], param_value).make_unique_id() else: params_str[param_name] = params[param_name].serialize(param_value) return params_str