Fix issue with test

sony · Nov 22, 2023 · a4054cc · a4054cc
1 parent 6394771
commit a4054cc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/keras_tests/function_tests/test_hessian_info_calculator.py b/tests/keras_tests/function_tests/test_hessian_info_calculator.py
@@ -48,6 +48,7 @@ def reused_model(input_shape):
     inputs = Input(shape=input_shape[1:])
     x = reused_layer(inputs)
     x = reused_layer(x)
+    x = ReLU()(x)
     return keras.Model(inputs=inputs, outputs=x)