Skip to content

Commit 3803327

Browse files
committed
rename variable [no ci]
1 parent ecd7b09 commit 3803327

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

bayesflow/diagnostics/plots/calibration_ecdf.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -146,11 +146,11 @@ def calibration_ecdf(
146146
test_quantities_targets[key] = np.expand_dims(tq_targets, axis=1)
147147

148148
# We assume test_quantity_func can only handle a 1D batch_size, so estimates
149-
# which have shape (num_conditions, num_post_samples, ...) must be flattend first.
150-
num_conditions, num_post_samples = next(iter(estimates.values())).shape[:2]
149+
# which have shape (num_conditions, num_samples, ...) must be flattend first.
150+
num_conditions, num_samples = next(iter(estimates.values())).shape[:2]
151151
flattened_estimates = keras.tree.map_structure(lambda t: np.reshape(t, (-1, *t.shape[2:])), estimates)
152152
flat_tq_estimates = test_quantity_func(data=flattened_estimates)
153-
test_quantities_estimates[key] = np.reshape(flat_tq_estimates, (num_conditions, num_post_samples, 1))
153+
test_quantities_estimates[key] = np.reshape(flat_tq_estimates, (num_conditions, num_samples, 1))
154154

155155
# Add custom test quantities to variable keys and names for plotting
156156
variable_keys = list(test_quantities.keys()) + variable_keys

0 commit comments

Comments
 (0)