Skip to content

Commit 1d7cb0f

Browse files
committed
fix dict key naming inconsistency ('targets' -> 'parameters') [skip ci]
1 parent 60cb270 commit 1d7cb0f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesflow/experimental/rectifiers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def call(self, input_dict, return_summary=False, num_eval_points=32, **kwargs):
207207
)
208208

209209
# Extract target variables
210-
target_vars = input_dict[defaults.DEFAULT_KEYS["targets"]]
210+
target_vars = input_dict[defaults.DEFAULT_KEYS["parameters"]]
211211

212212
# Extract batch size (autograph friendly)
213213
batch_size = tf.shape(target_vars)[0]
@@ -327,7 +327,7 @@ def log_density(self, input_dict, to_numpy=True, step_size=1e-3, **kwargs):
327327
)
328328

329329
# Extract targets
330-
target_vars = input_dict[defaults.DEFAULT_KEYS["targets"]]
330+
target_vars = input_dict[defaults.DEFAULT_KEYS["parameters"]]
331331

332332
# Reverse ODE and log pdf computation with the trace method
333333
latents, trace = self._solve_euler_inv(target_vars, conditions, step_size, **kwargs)

0 commit comments

Comments
 (0)