From 675652e43dfb7d89b5c21b7b3e0ecd03d84460fc Mon Sep 17 00:00:00 2001 From: Abhinav Singh Date: Tue, 22 Aug 2023 22:55:30 +0530 Subject: [PATCH] removing tf.keras dependency (#771) * removing tf.keras dependency * removing tf.keras dependency * removing tf.keras dependency * removing tf.keras dependency --- keras_core/layers/preprocessing/hashed_crossing_test.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/keras_core/layers/preprocessing/hashed_crossing_test.py b/keras_core/layers/preprocessing/hashed_crossing_test.py index fb97e5bcd..992ca25d0 100644 --- a/keras_core/layers/preprocessing/hashed_crossing_test.py +++ b/keras_core/layers/preprocessing/hashed_crossing_test.py @@ -145,9 +145,7 @@ def test_tf_string(self): outputs = layer((feat1, feat2)) self.assertAllClose(outputs, 1) - layer = tf.keras.layers.HashedCrossing( - num_bins=5, output_mode="one_hot" - ) + layer = layers.HashedCrossing(num_bins=5, output_mode="one_hot") feat1 = tf.constant(["A", "B", "A", "B", "A"]) feat2 = tf.constant([101, 101, 101, 102, 102]) self.assertAllClose( @@ -163,7 +161,7 @@ def test_tf_string(self): layer((feat1, feat2)), ) - layer = tf.keras.layers.HashedCrossing(num_bins=5) + layer = layers.HashedCrossing(num_bins=5) feat1 = tf.constant(["A", "B", "A", "B", "A"]) feat2 = tf.constant([101, 101, 101, 102, 102]) self.assertAllClose(tf.constant([1, 4, 1, 1, 3]), layer((feat1, feat2)))