how to add addiontional paramerters to Tasks

[puraminy]

I want to specify that columns used as prefix, input and target for a task,
I tried:

 t5.data.TaskRegistry.add(
                    f"{model}{lang}2en_{rel}{natural}_task",
                    # Specify the task type.
                    core.TvsTask,
                    # The followings are my columns
                    sel_cols = ["prefix", f"natural_input_text_{lang}", "target_text"],
                    # Supply a function which returns a tf.data.Dataset.
                    split_to_filepattern=paths,
                    ........

And in my TvsTask class I did:

class TvsTask(t5.data.Task):
   def __init__(
        self,
        name: str,
        sel_cols: List,
        split_to_filepattern: Dict[str, str],
        num_input_examples: Dict[str, int],
        text_preprocessor: Union[Callable, Sequence[Callable]],
        metric_fns: Sequence[Callable],
        truncate_to: Optional[int] = None,
        **kwargs,
    ) -> None:
        self.split_paths = split_to_filepattern
        self.exampes_nums = num_input_examples
        self.sel_cols = sel_cols,

However, I get the following error:

Traceback (most recent call last):
  File "/home/pouramini/rainbow/bin/list-tasks", line 7, in <module>
    import rainbow.tasks
  File "/home/pouramini/rainbow/src/rainbow/__init__.py", line 3, in <module>
    from . import (
  File "/home/pouramini/rainbow/src/rainbow/tasks.py", line 59, in <module>
    output_features=MT5_OUTPUT_FEATURES #if lang == "per" else None
  File "/home/pouramini/text-to-text-transfer-transformer/t5/data/dataset_providers.py", lin
e 293, in add
    provider = task_cls(name, **kwargs)
TypeError: __init__() missing 1 required positional argument: 'sel_cols'

Are there more standard way to do that? What other changes I must do to force my new arguments to a task?

[puraminy]

The error was from other lines and now it was resolved

[puraminy]

However, if there are still more standard solutions for what I want to do, please guide me.