diff --git a/seqio/__init__.py b/seqio/__init__.py
index 4dd594e1..98743c27 100644
--- a/seqio/__init__.py
+++ b/seqio/__init__.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/beam_utils.py b/seqio/beam_utils.py
index d9515a99..c2b6f4f8 100644
--- a/seqio/beam_utils.py
+++ b/seqio/beam_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/beam_utils_test.py b/seqio/beam_utils_test.py
index 3b9b2199..a4c271fe 100644
--- a/seqio/beam_utils_test.py
+++ b/seqio/beam_utils_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/benchmarks/device_benchmark.py b/seqio/benchmarks/device_benchmark.py
index c4cdd829..0aea98f8 100644
--- a/seqio/benchmarks/device_benchmark.py
+++ b/seqio/benchmarks/device_benchmark.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/benchmarks/preprocessors_benchmark.py b/seqio/benchmarks/preprocessors_benchmark.py
index 92a1610b..c16011d1 100644
--- a/seqio/benchmarks/preprocessors_benchmark.py
+++ b/seqio/benchmarks/preprocessors_benchmark.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/dataset_providers.py b/seqio/dataset_providers.py
index 397f4e48..4ad6383a 100644
--- a/seqio/dataset_providers.py
+++ b/seqio/dataset_providers.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -583,7 +583,7 @@ class FileDataSource(DataSource):
 
   def __init__(
       self,
-      read_file_fn: Callable[[tf.data.Dataset], tf.data.Dataset],
+      read_file_fn: Callable[[str], tf.data.Dataset],
       split_to_filepattern: Mapping[str, Union[str, Iterable[str]]],
       num_input_examples: Optional[Mapping[str, int]] = None,
       caching_permitted: bool = True,
diff --git a/seqio/dataset_providers_helpers.py b/seqio/dataset_providers_helpers.py
index 4f59c7da..adee7198 100644
--- a/seqio/dataset_providers_helpers.py
+++ b/seqio/dataset_providers_helpers.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/dataset_providers_test.py b/seqio/dataset_providers_test.py
index 3b61b991..d8009598 100644
--- a/seqio/dataset_providers_test.py
+++ b/seqio/dataset_providers_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/docs/conf.py b/seqio/docs/conf.py
index ef4d5cba..a08eacb1 100644
--- a/seqio/docs/conf.py
+++ b/seqio/docs/conf.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/docs/conf_sphinx_patch.py b/seqio/docs/conf_sphinx_patch.py
index 8fc3fdad..9b88d963 100644
--- a/seqio/docs/conf_sphinx_patch.py
+++ b/seqio/docs/conf_sphinx_patch.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/evaluation.py b/seqio/evaluation.py
index 5231199f..0148a659 100644
--- a/seqio/evaluation.py
+++ b/seqio/evaluation.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/evaluation_test.py b/seqio/evaluation_test.py
index 80eab2f5..f56aabb9 100644
--- a/seqio/evaluation_test.py
+++ b/seqio/evaluation_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/experimental.py b/seqio/experimental.py
index 77f3907e..15262cf1 100644
--- a/seqio/experimental.py
+++ b/seqio/experimental.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/experimental_test.py b/seqio/experimental_test.py
index 2f08521e..c1014ed8 100644
--- a/seqio/experimental_test.py
+++ b/seqio/experimental_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/feature_converters.py b/seqio/feature_converters.py
index 526e69e7..3e64be16 100644
--- a/seqio/feature_converters.py
+++ b/seqio/feature_converters.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/feature_converters_test.py b/seqio/feature_converters_test.py
index d2e73139..befef1e8 100644
--- a/seqio/feature_converters_test.py
+++ b/seqio/feature_converters_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/helpers.py b/seqio/helpers.py
index 04ba0c5e..719e194b 100644
--- a/seqio/helpers.py
+++ b/seqio/helpers.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/helpers_test.py b/seqio/helpers_test.py
index 42c49b00..25ff342b 100644
--- a/seqio/helpers_test.py
+++ b/seqio/helpers_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/loggers.py b/seqio/loggers.py
index 37b0667b..9e659d09 100644
--- a/seqio/loggers.py
+++ b/seqio/loggers.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/loggers_test.py b/seqio/loggers_test.py
index 5e42ced6..5e25ecbd 100644
--- a/seqio/loggers_test.py
+++ b/seqio/loggers_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/metrics.py b/seqio/metrics.py
index 7a889a68..601f0d98 100644
--- a/seqio/metrics.py
+++ b/seqio/metrics.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/metrics_test.py b/seqio/metrics_test.py
index 139272d5..1adcf97e 100644
--- a/seqio/metrics_test.py
+++ b/seqio/metrics_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/preprocessors.py b/seqio/preprocessors.py
index 99d28c3a..bd96fa37 100644
--- a/seqio/preprocessors.py
+++ b/seqio/preprocessors.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/preprocessors_test.py b/seqio/preprocessors_test.py
index 22b0f1c7..15173145 100644
--- a/seqio/preprocessors_test.py
+++ b/seqio/preprocessors_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/scripts/__init__.py b/seqio/scripts/__init__.py
index 1fdfa292..23f668a4 100644
--- a/seqio/scripts/__init__.py
+++ b/seqio/scripts/__init__.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/scripts/cache_tasks_main.py b/seqio/scripts/cache_tasks_main.py
index 6c71e479..2cef90b2 100644
--- a/seqio/scripts/cache_tasks_main.py
+++ b/seqio/scripts/cache_tasks_main.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/scripts/cache_tasks_test.py b/seqio/scripts/cache_tasks_test.py
index 81dcf3ee..11ecc065 100644
--- a/seqio/scripts/cache_tasks_test.py
+++ b/seqio/scripts/cache_tasks_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/scripts/inspect_tasks_main.py b/seqio/scripts/inspect_tasks_main.py
index 6c41d02a..b8605cf6 100644
--- a/seqio/scripts/inspect_tasks_main.py
+++ b/seqio/scripts/inspect_tasks_main.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/scripts/registration_provenance_main.py b/seqio/scripts/registration_provenance_main.py
index d82dab7b..cf7592dd 100644
--- a/seqio/scripts/registration_provenance_main.py
+++ b/seqio/scripts/registration_provenance_main.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/scripts/registration_provenance_test.py b/seqio/scripts/registration_provenance_test.py
index f1b3b1ee..62ad81ca 100644
--- a/seqio/scripts/registration_provenance_test.py
+++ b/seqio/scripts/registration_provenance_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/sharded_evaluation_test.py b/seqio/sharded_evaluation_test.py
index a1c72bbc..0268db63 100644
--- a/seqio/sharded_evaluation_test.py
+++ b/seqio/sharded_evaluation_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/task_registry_provenance_tracking.py b/seqio/task_registry_provenance_tracking.py
index e120bfb1..2bdfe2f4 100644
--- a/seqio/task_registry_provenance_tracking.py
+++ b/seqio/task_registry_provenance_tracking.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/task_registry_provenance_tracking_test.py b/seqio/task_registry_provenance_tracking_test.py
index b6db93c3..c7e5e51f 100644
--- a/seqio/task_registry_provenance_tracking_test.py
+++ b/seqio/task_registry_provenance_tracking_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/test_utils.py b/seqio/test_utils.py
index fe8bb5d7..697702a2 100644
--- a/seqio/test_utils.py
+++ b/seqio/test_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/test_utils_test.py b/seqio/test_utils_test.py
index a23427b2..09287407 100644
--- a/seqio/test_utils_test.py
+++ b/seqio/test_utils_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/utils.py b/seqio/utils.py
index f3502b98..87190243 100644
--- a/seqio/utils.py
+++ b/seqio/utils.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/utils_test.py b/seqio/utils_test.py
index e91c0f5a..5154e06f 100644
--- a/seqio/utils_test.py
+++ b/seqio/utils_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/version.py b/seqio/version.py
index db9836a5..3df06665 100644
--- a/seqio/version.py
+++ b/seqio/version.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/vocabularies.py b/seqio/vocabularies.py
index 8057ef06..bf9256ac 100644
--- a/seqio/vocabularies.py
+++ b/seqio/vocabularies.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/seqio/vocabularies_test.py b/seqio/vocabularies_test.py
index c57c9273..de513cad 100644
--- a/seqio/vocabularies_test.py
+++ b/seqio/vocabularies_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/setup.py b/setup.py
index f279ede4..1f556d14 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The SeqIO Authors.
+# Copyright 2024 The SeqIO Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.