From ce90d8fc0e9dcb6c9693be42d9a6d19681719ea2 Mon Sep 17 00:00:00 2001
From: Tom van der Weide <weide@google.com>
Date: Fri, 3 Feb 2023 02:36:26 -0800
Subject: [PATCH] Internal change

PiperOrigin-RevId: 506850267
---
 seqio/dataset_providers.py | 1 +
 1 file changed, 1 insertion(+)

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,