diff --git a/bayesflow/approximators/continuous_approximator.py b/bayesflow/approximators/continuous_approximator.py index 13ba32cb9..c34bcc908 100644 --- a/bayesflow/approximators/continuous_approximator.py +++ b/bayesflow/approximators/continuous_approximator.py @@ -570,11 +570,11 @@ def summarize(self, data: Mapping[str, np.ndarray], **kwargs) -> np.ndarray: if self.summary_network is None: raise ValueError("A summary network is required to compute summaries.") - data_adapted = self.adapter(data, strict=False, **kwargs) + data_adapted = self._prepare_data(data, **kwargs) if "summary_variables" not in data_adapted or data_adapted["summary_variables"] is None: raise ValueError("Summary variables are required to compute summaries.") - summary_variables = keras.tree.map_structure(keras.ops.convert_to_tensor, data_adapted["summary_variables"]) + summary_variables = data_adapted["summary_variables"] summaries = self.summary_network(summary_variables, **filter_kwargs(kwargs, self.summary_network.call)) summaries = keras.ops.convert_to_numpy(summaries) diff --git a/bayesflow/approximators/model_comparison_approximator.py b/bayesflow/approximators/model_comparison_approximator.py index d71aafdaf..034281695 100644 --- a/bayesflow/approximators/model_comparison_approximator.py +++ b/bayesflow/approximators/model_comparison_approximator.py @@ -433,6 +433,10 @@ def summarize(self, data: Mapping[str, np.ndarray], **kwargs) -> np.ndarray: raise ValueError("Summary variables are required to compute summaries.") summary_variables = keras.tree.map_structure(keras.ops.convert_to_tensor, data_adapted["summary_variables"]) + + if "summary_variables" in self.standardize: + summary_variables = self.standardize_layers["summary_variables"](summary_variables) + summaries = self.summary_network(summary_variables, **filter_kwargs(kwargs, self.summary_network.call)) summaries = keras.ops.convert_to_numpy(summaries)