removed unnecessary kwarg

2022-03-27 01:24:15 +01:00 · 2022-03-27 01:24:15 +01:00 · 0f811bdc56
commit 0f811bdc56
parent d11333981f
1 changed files with 1 additions and 1 deletions
--- a/test/unit_tests/conftest.py
+++ b/test/unit_tests/conftest.py
@ -58,7 +58,7 @@ def keras_model(input_size):

    inputs = tf.keras.Input(shape=input_size)
    conv = tf.keras.layers.Conv2D(3, 3)
-    dense = tf.keras.layers.Dense(10, activation="relu")
+    dense = tf.keras.layers.Dense(10)

    outputs = tf.keras.layers.Dense(10)(dense(conv(inputs)))
    model = tf.keras.Model(inputs=inputs, outputs=outputs)