From ac7b66ef0eedceb72c456067aaf0da855099e2e4 Mon Sep 17 00:00:00 2001 From: stefanradev93 Date: Fri, 17 May 2024 13:52:29 -0400 Subject: [PATCH] Correctly propagate kwargs in constructor --- .../experimental/networks/coupling_flow/couplings/coupling.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bayesflow/experimental/networks/coupling_flow/couplings/coupling.py b/bayesflow/experimental/networks/coupling_flow/couplings/coupling.py index 0167a258d..12fe25904 100644 --- a/bayesflow/experimental/networks/coupling_flow/couplings/coupling.py +++ b/bayesflow/experimental/networks/coupling_flow/couplings/coupling.py @@ -15,14 +15,14 @@ def __init__( transform: str, **kwargs ): - super().__init__(**kwargs) + super().__init__() self.half_dim = half_dim self.subnet = find_subnet( subnet=subnet_builder, transform=transform, output_dim=half_dim, - **kwargs.pop('subnet_kwargs', {}) + **kwargs.pop('subnet_settings', {}) ) self.transform = find_transform(transform)