diff --git a/seqio/dataset_providers.py b/seqio/dataset_providers.py index 2a20570e..435aac7a 100644 --- a/seqio/dataset_providers.py +++ b/seqio/dataset_providers.py @@ -1433,6 +1433,7 @@ class TaskRegistry(DatasetProviderRegistry): # pylint: disable=arguments-renamed @classmethod + @seqio_logger.register_task() def add( cls, name: str,