From 6d827e6f7d74e3a5a527381c7cd05fa510a13d53 Mon Sep 17 00:00:00 2001 From: RollerKnobster Date: Wed, 26 Jun 2024 10:20:26 +0300 Subject: [PATCH] Fix: add input_shape to Dense layers in kerasraw test --- tests/gordo/machine/model/test_raw_keras.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/gordo/machine/model/test_raw_keras.py b/tests/gordo/machine/model/test_raw_keras.py index f6a8290a6..e3f179196 100644 --- a/tests/gordo/machine/model/test_raw_keras.py +++ b/tests/gordo/machine/model/test_raw_keras.py @@ -81,8 +81,12 @@ def test_raw_keras_part_of_pipeline(): layers: - tensorflow.keras.layers.Dense: units: 4 + input_shape: + - 4 - tensorflow.keras.layers.Dense: units: 1 + input_shape: + - 1 """ config = yaml.safe_load(config_str) pipe = serializer.from_definition(config)