Merge pull request #1338 from gatl/error_message

Improve error message
EpistasisLab · Feb 22, 2024 · 3fd4e9b · 3fd4e9b
2 parents db64799 + bf56049
commit 3fd4e9b
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 "