Skip to content

Commit 48bb190

Browse files
committed
Fix in model comparison
1 parent 333c30f commit 48bb190

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bayesflow/approximators/model_comparison_approximator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def build(self, data_shapes: dict[str, tuple[int] | dict[str, dict]]) -> None:
9292

9393
# Set up standardization layers if requested
9494
if self.standardize == "all":
95-
self.standardize = [var for var in ["summary_variables", "classifier_conditions"] if var in data_shapes]
95+
self.standardize = [var for var in ModelComparisonApproximator.CONDITION_KEYS if var in data_shapes]
9696
self.standardize_layers = {var: Standardization(trainable=False) for var in self.standardize}
9797

9898
# Build all standardization layers

0 commit comments

Comments
 (0)