Revert part of some breaking changes in using rate_per_task_name

PiperOrigin-RevId: 596903315
google · Jan 9, 2024 · 6df1412 · 6df1412
1 parent 19e6aec
commit 6df1412
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/seqio/experimental.py b/seqio/experimental.py
@@ -206,7 +206,7 @@ def add_fully_cached_mixture(
   return MixtureRegistry.add(
       new_name,
       [
-          (new_t.name, mixture.rate_per_task_name[old_t.name])
+          (new_t.name, mixture._task_to_rate[old_t.name])  # pylint:disable=protected-access
           for old_t, new_t in zip(mixture.tasks, new_tasks)
       ],
   )