From 50889fe14582df77b417e8402e4a20be90b30112 Mon Sep 17 00:00:00 2001 From: Joan Puigcerver Date: Fri, 5 Aug 2022 02:59:14 -0700 Subject: [PATCH] Internal changes. PiperOrigin-RevId: 465522238 --- vmoe/data/input_pipeline.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vmoe/data/input_pipeline.py b/vmoe/data/input_pipeline.py index 4ffda0f..cec5ffa 100644 --- a/vmoe/data/input_pipeline.py +++ b/vmoe/data/input_pipeline.py @@ -20,6 +20,7 @@ import ast from typing import Any, Callable, Dict, Optional, Union +from absl import logging import clu.data import jax import ml_collections