Skip to content

Commit a4cbd16

Browse files
committed
fix flow matching kwargs updating
1 parent c313871 commit a4cbd16

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesflow/networks/flow_matching/flow_matching.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,11 +67,11 @@ def __init__(
6767
self.use_optimal_transport = use_optimal_transport
6868

6969
new_integrate_kwargs = FlowMatching.INTEGRATE_DEFAULT_CONFIG.copy()
70-
new_integrate_kwargs.update(integrate_kwargs)
70+
new_integrate_kwargs.update(integrate_kwargs or {})
7171
self.integrate_kwargs = new_integrate_kwargs
7272

7373
new_optimal_transport_kwargs = FlowMatching.OPTIMAL_TRANSPORT_DEFAULT_CONFIG.copy()
74-
new_optimal_transport_kwargs.update(optimal_transport_kwargs)
74+
new_optimal_transport_kwargs.update(optimal_transport_kwargs or {})
7575
self.optimal_transport_kwargs = new_optimal_transport_kwargs
7676

7777
self.loss_fn = keras.losses.get(loss_fn)

0 commit comments

Comments
 (0)