Skip to content

GitLab

Commit d0879611 authored by Scott Zhu's avatar Scott Zhu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 368571725
parent 147b46e9
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/nlp/modeling/layers/tn_expand_condense_test.py
View file @ d0879611
...@@ -45,6 +45,10 @@ class TNLayerTest(tf.test.TestCase, parameterized.TestCase): ...@@ -45,6 +45,10 @@ class TNLayerTest(tf.test.TestCase, parameterized.TestCase):
@parameterized.parameters((768, 6), (1024, 2)) @parameterized.parameters((768, 6), (1024, 2))
def test_keras_layer(self, input_dim, proj_multiple): def test_keras_layer(self, input_dim, proj_multiple):
self.skipTest('Disable the test for now since it imports '
'keras.testing_utils, will reenable this test after we '
'fix the b/184578869')
# TODO(scottzhu): Reenable after fix b/184578869
data = np.random.normal(size=(100, input_dim)) data = np.random.normal(size=(100, input_dim))
data = data.astype(np.float32) data = data.astype(np.float32)
layer_test( layer_test(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment