fix integrate_kwargs

arrjon · arrjon · commit 0ff960f8dfd8 · 2025-09-13T13:59:01.000+02:00
diff --git a/bayesflow/networks/diffusion_model/diffusion_model.py b/bayesflow/networks/diffusion_model/diffusion_model.py
@@ -812,7 +812,7 @@ def _inverse_compositional(
                     "or use another backend."
                 )
         else:
-            mini_batch_size = integrate_kwargs.get("mini_batch_size", int(n_compositional * 0.1))
+            mini_batch_size = integrate_kwargs.pop("mini_batch_size", int(n_compositional * 0.1))
         self.compositional_d0 = float(integrate_kwargs.pop("compositional_d0", 1.0))
         self.compositional_d1 = float(integrate_kwargs.pop("compositional_d1", 1.0))
 

Original file line number	Diff line number	Diff line change
`@@ -812,7 +812,7 @@ def _inverse_compositional(`
`812`	`812`	`"or use another backend."`
`813`	`813`	`)`
`814`	`814`	`else:`
`815`		`- mini_batch_size = integrate_kwargs.get("mini_batch_size", int(n_compositional * 0.1))`
	`815`	`+ mini_batch_size = integrate_kwargs.pop("mini_batch_size", int(n_compositional * 0.1))`
`816`	`816`	`self.compositional_d0 = float(integrate_kwargs.pop("compositional_d0", 1.0))`
`817`	`817`	`self.compositional_d1 = float(integrate_kwargs.pop("compositional_d1", 1.0))`
`818`	`818`