fix dict key naming inconsistency ('targets' -> 'parameters') [skip ci]

bayesflow-org · Jul 28, 2023 · 1d7cb0f · 1d7cb0f
1 parent 60cb270
commit 1d7cb0f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bayesflow/experimental/rectifiers.py b/bayesflow/experimental/rectifiers.py
@@ -207,7 +207,7 @@ def call(self, input_dict, return_summary=False, num_eval_points=32, **kwargs):
         )
 
         # Extract target variables
-        target_vars = input_dict[defaults.DEFAULT_KEYS["targets"]]
+        target_vars = input_dict[defaults.DEFAULT_KEYS["parameters"]]
 
         # Extract batch size (autograph friendly)
         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):
         )
 
         # Extract targets
-        target_vars = input_dict[defaults.DEFAULT_KEYS["targets"]]
+        target_vars = input_dict[defaults.DEFAULT_KEYS["parameters"]]
 
         # Reverse ODE and log pdf computation with the trace method
         latents, trace = self._solve_euler_inv(target_vars, conditions, step_size, **kwargs)