internal

PiperOrigin-RevId: 542322982
google-research · Jul 13, 2023 · db3a4e3 · db3a4e3
1 parent 2e1dbce
commit db3a4e3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/t5x/test_utils.py b/t5x/test_utils.py
@@ -26,6 +26,7 @@
 from jax.sharding import Mesh
 import numpy as np
 import seqio
+import t5.data
 from t5x import adafactor
 from t5x import models
 from t5x import partitioning