From 7f2c00913a32fcc4d09db89c51bb86d6fe1a59e8 Mon Sep 17 00:00:00 2001 From: LysandreJik Date: Mon, 23 Nov 2020 18:19:54 -0500 Subject: [PATCH] TF BERT test update --- tests/test_modeling_tf_bert.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_modeling_tf_bert.py b/tests/test_modeling_tf_bert.py index 64747f5d35..cda2157f0f 100644 --- a/tests/test_modeling_tf_bert.py +++ b/tests/test_modeling_tf_bert.py @@ -356,9 +356,9 @@ class TFBertModelIntegrationTest(unittest.TestCase): expected_slice = tf.constant( [ [ - [-0.05243197, -0.04498899, 0.05512108], - [-0.07444685, -0.01064632, 0.04352357], - [-0.05020351, 0.05530146, 0.00700043] + [-0.05243197, -0.04498899, 0.05512108], + [-0.07444685, -0.01064632, 0.04352357], + [-0.05020351, 0.05530146, 0.00700043], ] ] )