Compatibility: deserialize when get_config was overridden

Author: vpratz

bayesflow/networks/summary_network.py

@@ -4,6 +4,7 @@
 from bayesflow.types import Tensor
 from bayesflow.utils import layer_kwargs, find_distribution
 from bayesflow.utils.decorators import sanitize_input_shape
+from bayesflow.utils.serialization import deserialize
 
 
 class SummaryNetwork(keras.Layer):
@@ -49,3 +50,9 @@ def compute_metrics(self, x: Tensor, stage: str = "training", **kwargs) -> dict[
                     metrics[metric.name] = metric(outputs, samples)
 
         return metrics
+
+    @classmethod
+    def from_config(cls, config, custom_objects=None):
+        if hasattr(cls.get_config, "_is_default") and cls.get_config._is_default:
+            return cls(**config)
+        return cls(**deserialize(config, custom_objects=custom_objects))