From 8e8b05ba12884e6deb689073ae79806466003832 Mon Sep 17 00:00:00 2001 From: RecML authors Date: Mon, 3 Aug 2026 17:32:31 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 958666981 --- recml/core/utils/keras_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/recml/core/utils/keras_utils.py b/recml/core/utils/keras_utils.py index 1109a8f..31eb566 100644 --- a/recml/core/utils/keras_utils.py +++ b/recml/core/utils/keras_utils.py @@ -21,6 +21,7 @@ from typing import Any from absl import logging + import jax import keras import orbax.checkpoint as ocp