Remove the build deps to legacy keras code in tensorflow/python/keras.

PiperOrigin-RevId: 495929362

Author: qlzh727

tensorflow_model_optimization/python/core/keras/testing/BUILD

@@ -15,6 +15,5 @@ py_strict_library(
     deps = [
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
     ],
 )

tensorflow_model_optimization/python/core/keras/testing/test_utils_mnist.py

@@ -16,9 +16,9 @@
 
 import numpy as np
 import tensorflow as tf
+from tensorflow import keras
 
-from tensorflow.python import keras
-l = keras.layers
+l = tf.keras.layers
 
 
 def layers_list():

tensorflow_model_optimization/python/core/quantization/keras/BUILD

@@ -45,7 +45,6 @@ py_strict_test(
         ":quant_ops",
         # absl/testing:parameterized dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:compat",
     ],
 )
@@ -76,7 +75,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:compat",
     ],
 )
@@ -172,7 +170,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
     ],
 )
 
@@ -292,7 +289,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:compat",
         "//tensorflow_model_optimization/python/core/keras:test_utils",
     ],
@@ -311,7 +307,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
     ],
 )
 
@@ -328,7 +323,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:compat",
         "//tensorflow_model_optimization/python/core/keras/testing:test_utils_mnist",
     ],

tensorflow_model_optimization/python/core/sparsity/keras/BUILD

@@ -88,7 +88,6 @@ py_strict_test(
         ":pruning_schedule",
         # absl/testing:parameterized dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:compat",
     ],
 )
@@ -139,7 +138,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:test_utils",
     ],
 )
@@ -209,7 +207,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:test_utils",
     ],
 )
@@ -231,7 +228,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:test_utils",
     ],
 )
@@ -295,7 +291,6 @@ py_strict_test(
         # absl/testing:parameterized dep1,
         # numpy dep1,
         # tensorflow dep1,
-        # python/keras tensorflow dep2,
         "//tensorflow_model_optimization/python/core/keras:compat",
     ],
 )