diff --git a/khiops/sklearn/estimators.py b/khiops/sklearn/estimators.py index c712060d..a427d230 100644 --- a/khiops/sklearn/estimators.py +++ b/khiops/sklearn/estimators.py @@ -1435,7 +1435,11 @@ def _fit_training_post_process(self, ds): else: pair_feature_evaluated_names_ = [] pair_feature_evaluated_levels_ = [] - if "treePreparationReport" in self.model_report_.json_data: + if ( + "treePreparationReport" in self.model_report_.json_data + and "variablesStatistics" + in self.model_report_.json_data["treePreparationReport"] + ): tree_preparation_report = self.model_report_.json_data[ "treePreparationReport" ]["variablesStatistics"]