From 8fb60e11433517300d41b9b37a8e923497f4ea96 Mon Sep 17 00:00:00 2001 From: Hao Zhou Date: Tue, 6 Feb 2024 11:58:48 -0800 Subject: [PATCH] Add import version check for tf keras module. Make sure TFMA depend on keras 2 to fix the metrics module mismatch when using TFMA with keras metrics. PiperOrigin-RevId: 604720506 --- tf_keras/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tf_keras/__init__.py b/tf_keras/__init__.py index 34719ad43..326cb731e 100644 --- a/tf_keras/__init__.py +++ b/tf_keras/__init__.py @@ -24,6 +24,7 @@ from tf_keras import losses from tf_keras import metrics from tf_keras import models +from tf_keras import optimizers from tf_keras.engine.input_layer import Input from tf_keras.engine.sequential import Sequential from tf_keras.engine.training import Model