Skip to content

Commit b7d6c0e

Browse files
committed
fix sample post-processing in point approximator
1 parent e32ae2e commit b7d6c0e

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

bayesflow/approximators/point_approximator.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,6 @@ def sample(
133133
for score_key in samples.keys():
134134
samples[score_key] = self.standardize_layers["inference_variables"](samples[score_key], forward=False)
135135

136-
samples = {"inference_variables": samples}
137136
samples = self._apply_inverse_adapter_to_samples(samples, **kwargs)
138137

139138
if split:
@@ -218,9 +217,9 @@ def _apply_inverse_adapter_to_samples(
218217
"""Applies the inverse adapter to a dictionary of samples."""
219218
samples = keras.tree.map_structure(keras.ops.convert_to_numpy, samples)
220219
processed = {}
221-
for score_key, samples in samples.items():
220+
for score_key, score_value in samples.items():
222221
processed[score_key] = self.adapter(
223-
{"inference_variables": samples},
222+
{"inference_variables": score_value},
224223
inverse=True,
225224
strict=False,
226225
**kwargs,

0 commit comments

Comments
 (0)