Skip to content

Commit 675652e

Browse files
authored
removing tf.keras dependency (#771)
* removing tf.keras dependency * removing tf.keras dependency * removing tf.keras dependency * removing tf.keras dependency
1 parent e6a39ba commit 675652e

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

keras_core/layers/preprocessing/hashed_crossing_test.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -145,9 +145,7 @@ def test_tf_string(self):
145145
outputs = layer((feat1, feat2))
146146
self.assertAllClose(outputs, 1)
147147

148-
layer = tf.keras.layers.HashedCrossing(
149-
num_bins=5, output_mode="one_hot"
150-
)
148+
layer = layers.HashedCrossing(num_bins=5, output_mode="one_hot")
151149
feat1 = tf.constant(["A", "B", "A", "B", "A"])
152150
feat2 = tf.constant([101, 101, 101, 102, 102])
153151
self.assertAllClose(
@@ -163,7 +161,7 @@ def test_tf_string(self):
163161
layer((feat1, feat2)),
164162
)
165163

166-
layer = tf.keras.layers.HashedCrossing(num_bins=5)
164+
layer = layers.HashedCrossing(num_bins=5)
167165
feat1 = tf.constant(["A", "B", "A", "B", "A"])
168166
feat2 = tf.constant([101, 101, 101, 102, 102])
169167
self.assertAllClose(tf.constant([1, 4, 1, 1, 3]), layer((feat1, feat2)))

0 commit comments

Comments
 (0)