Improve error message

Unfortunately there is no fine granularity when an error occurs. That makes it more relevant to list as many possibilities for the causes as possible. In this patch, one additional cause is added to the ones already presented (no optimization done due to timeout from max_eval_time_mins).
EpistasisLab · Dec 12, 2023 · bf56049 · bf56049
1 parent 71933de
commit bf56049
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tpot/base.py b/tpot/base.py
@@ -935,7 +935,8 @@ def _update_top_pipeline(self):
                 raise RuntimeError(
                     "There was an error in the TPOT optimization "
                     "process. This could be because the data was "
-                    "not formatted properly, or because data for "
+                    "not formatted properly, because a timeout "
+                    "was reached or because data for "
                     "a regression problem was provided to the "
                     "TPOTClassifier object. Please make sure you "
                     "passed the data to TPOT correctly. If you "