Fix: add input_shape to Dense layers in kerasraw test

equinor · Jun 26, 2024 · 6d827e6 · 6d827e6
1 parent 6852cbc
commit 6d827e6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)