fix ConcatenateMLP

Author: arrjon

tests/test_networks/conftest.py

@@ -37,15 +37,21 @@ def compute_output_shape(self, input_shape_x, input_shape_t, input_shape_conditi
                 + (input_shape_conditions[-1] if input_shape_conditions is not None else 0),
             )
         )
-        out = self.mlp.compute_output_shape(concatenate_input_shapes)
-        return out
+        return self.mlp.compute_output_shape(concatenate_input_shapes)
 
     def build(self, input_shape_x, input_shape_t, input_shape_conditions=None):
         if self.built:
             return
 
-        input_shape = self.compute_output_shape(input_shape_x, input_shape_t, input_shape_conditions)
-        self.mlp.build(input_shape)
+        concatenate_input_shapes = tuple(
+            (
+                input_shape_x[0],
+                input_shape_x[-1]
+                + input_shape_t[-1]
+                + (input_shape_conditions[-1] if input_shape_conditions is not None else 0),
+            )
+        )
+        self.mlp.build(concatenate_input_shapes)
 
 
 @pytest.fixture()