Skip to content

Conversation

@Innixma
Copy link
Contributor

@Innixma Innixma commented Feb 26, 2025

Issue #, if available:

Description of changes:

  • Add raise_on_model_failure flag (default False)
  • When True, will immediately raise the original exception if a model raises an exception during fit.
  • Setting to True is very helpful when using a debugger to try to figure out why a model is failing, as otherwise exceptions are handled by AutoGluon which isn't desired while debugging.
  • Docstring / type hint improvements

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@Innixma Innixma added this to the 1.3 Release milestone Feb 26, 2025
Copy link
Contributor

@canerturkmen canerturkmen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM overall. I would only encourage adding tests. If somebody else decides to catch the model errors (i.e., the learner starts catching stuff), the behavior will break.

This is useful when using a debugger during development to identify the cause of model failures.
This should only be used for debugging.
If False, will try to skip to the next model if an exception occurred during model training.
This is the default logic and is a core principal of AutoGluon's design.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

typo -> principle.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

updated

@Innixma
Copy link
Contributor Author

Innixma commented Feb 26, 2025

added unit test, good suggestion @canerturkmen !

@Innixma Innixma merged commit ff2178d into autogluon:master Feb 26, 2025
4 checks passed
@Innixma Innixma deleted the raise-on-model-failure branch April 16, 2025 21:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants