Simplify

bayesflow-org · Jun 28, 2024 · eae9814 · eae9814
1 parent 80695e1
commit eae9814
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/bayesflow/networks/mlp/mlp.py b/bayesflow/networks/mlp/mlp.py
@@ -76,11 +76,7 @@ def build(self, input_shape):
         # build nested layers with forward pass
         self.call(keras.ops.zeros(input_shape))
 
-    def call(self, inputs: Tensor, **kwargs) -> Tensor:
+    def call(self, x: Tensor, **kwargs) -> Tensor:
         for layer in self.res_blocks:
-            _kwargs = {}
-            if layer._call_has_training_arg:
-                _kwargs["training"] = kwargs.get("training", False)
-            outputs = layer(inputs, **kwargs)
-            inputs = outputs
-        return outputs
+            x = layer(x, training=kwargs.get("training", False))
+        return x