[no-ci] fix: filter kwargs of InferenceNetwork

Author: han-ol

bayesflow/networks/inference_network.py

@@ -1,15 +1,15 @@
 import keras
 
 from bayesflow.types import Shape, Tensor
-from bayesflow.utils import find_distribution
+from bayesflow.utils import find_distribution, keras_kwargs
 from bayesflow.utils.decorators import allow_batch_size
 
 
 class InferenceNetwork(keras.Layer):
     MLP_DEFAULT_CONFIG = {}
 
     def __init__(self, base_distribution: str = "normal", **kwargs):
-        super().__init__(**kwargs)
+        super().__init__(**keras_kwargs(kwargs))
         self.base_distribution = find_distribution(base_distribution)
 
     def build(self, xz_shape: Shape, conditions_shape: Shape = None) -> None: