Add keras:compat dependency to clustering callbacks.

PiperOrigin-RevId: 335719012
MatteoArm · Oct 6, 2020 · bba3f17 · bba3f17
1 parent 6b32758
commit bba3f17
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_model_optimization/python/core/clustering/keras/BUILD b/tensorflow_model_optimization/python/core/clustering/keras/BUILD
@@ -79,6 +79,7 @@ py_library(
     visibility = ["//visibility:public"],
     deps = [
         # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/keras:compat",
     ],
 )