fix: add the missing dot

boat-group · Feb 8, 2020 · cb954ee · cb954ee
1 parent 2c58213
commit cb954ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/fancy_nlp/layers/test_attention.py b/tests/fancy_nlp/layers/test_attention.py
@@ -7,6 +7,6 @@
 
 class TestAttention:
     def test_multihead_attention(self):
-        input_embed = tf.keras.layersInput(shape=(3, 300))
+        input_embed = tf.keras.layers.Input(shape=(3, 300))
         input_encode = MultiHeadAttention()(input_embed)
         model = tf.keras.models.Model(input_embed, input_encode)