autogluon.tabular 1.2.1b20250226__py3-none-any.whl → 1.2.1b20250227__py3-none-any.whl

autogluon/tabular/learner/abstract_learner.py

@@ -39,19 +39,19 @@ class AbstractTabularLearner(AbstractLearner):
         self,
         path_context: str,
         label: str,
-        feature_generator: PipelineFeatureGenerator,
+        feature_generator: PipelineFeatureGenerator | None = None,
         ignored_columns: list = None,
-        label_count_threshold=10,
-        problem_type=None,
-        quantile_levels=None,
-        eval_metric=None,
-        positive_class=None,
-        cache_data=True,
-        is_trainer_present=False,
-        random_state=0,
-        sample_weight=None,
-        weight_evaluation=False,
-        groups=None,
+        label_count_threshold: int = 10,
+        problem_type: str | None = None,
+        quantile_levels: list[float] | None = None,
+        eval_metric: Scorer | None = None,
+        positive_class: str | None = None,
+        cache_data: bool = True,
+        is_trainer_present: bool = False,
+        random_state: int = 0,
+        sample_weight: str | None = None,
+        weight_evaluation: bool = False,
+        groups: str | None = None,
     ):
         super().__init__(path_context=path_context, random_state=random_state)
         self.label = label

autogluon/tabular/learner/default_learner.py

@@ -43,16 +43,17 @@ class DefaultLearner(AbstractTabularLearner):
     def _fit(
         self,
         X: DataFrame,
-        X_val: DataFrame = None,
-        X_test: DataFrame = None,
-        X_unlabeled: DataFrame = None,
-        holdout_frac=0.1,
-        num_bag_folds=0,
-        num_bag_sets=1,
-        time_limit=None,
-        infer_limit=None,
-        infer_limit_batch_size=None,
-        verbosity=2,
+        X_val: DataFrame | None = None,
+        X_test: DataFrame | None = None,
+        X_unlabeled: DataFrame | None = None,
+        holdout_frac: float = 0.1,
+        num_bag_folds: int = 0,
+        num_bag_sets: int = 1,
+        time_limit: float | None = None,
+        infer_limit: float | None = None,
+        infer_limit_batch_size: int | None = None,
+        verbosity: int = 2,
+        raise_on_model_failure: bool = False,
         **trainer_fit_kwargs,
     ):
         """Arguments:
@@ -121,6 +122,7 @@ class DefaultLearner(AbstractTabularLearner):
             save_data=self.cache_data,
             random_state=self.random_state,
             verbosity=verbosity,
+            raise_on_model_failure=raise_on_model_failure,
         )
 
         self.trainer_path = trainer.path

autogluon/tabular/predictor/predictor.py

@@ -990,6 +990,14 @@ class TabularPredictor:
                 to any amount of labeled data.
             verbosity : int
                 If specified, overrides the existing `predictor.verbosity` value.
+            raise_on_model_failure: bool, default = False
+                If True, will raise on any exception during model training.
+                    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 principle of AutoGluon's design.
+
+                .. versionadded:: 1.3.0
             raise_on_no_models_fitted: bool, default = True
                 If True, will raise a RuntimeError if no models were successfully fit during `fit()`.
             calibrate: bool or str, default = 'auto'
@@ -1109,6 +1117,7 @@ class TabularPredictor:
         delay_bag_sets: bool = kwargs["delay_bag_sets"]
         test_data = kwargs["test_data"]
         learning_curves = kwargs["learning_curves"]
+        raise_on_model_failure = kwargs["raise_on_model_failure"]
 
         if ag_args is None:
             ag_args = {}
@@ -1256,6 +1265,7 @@ class TabularPredictor:
             verbosity=verbosity,
             use_bag_holdout=use_bag_holdout,
             callbacks=callbacks,
+            raise_on_model_failure=raise_on_model_failure,
         )
         ag_post_fit_kwargs = dict(
             keep_only_best=kwargs["keep_only_best"],
@@ -5036,6 +5046,7 @@ class TabularPredictor:
             # learning curves and test data (for logging purposes only)
             learning_curves=False,
             test_data=None,
+            raise_on_model_failure=False,
         )
         kwargs, ds_valid_keys = self._sanitize_dynamic_stacking_kwargs(kwargs)
         kwargs = self._validate_fit_extra_kwargs(kwargs, extra_valid_keys=list(fit_kwargs_default.keys()) + ds_valid_keys)

autogluon/tabular/trainer/abstract_trainer.py

@@ -128,6 +128,12 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         Higher levels correspond to more detailed print statements (you can set verbosity = 0 to suppress warnings).
         If using logging, you can alternatively control amount of information printed via `logger.setLevel(L)`,
         where `L` ranges from 0 to 50 (Note: higher values of `L` correspond to fewer print statements, opposite of verbosity levels).
+    raise_on_model_failure : bool, default = False
+        If True, Trainer will raise on any exception during model training.
+            This is ideal when using a debugger during development.
+        If False, Trainer will try to skip to the next model if an exception occurred during model training.
+
+        .. versionadded:: 1.3.0
     """
 
     distill_stackname = "distill"  # name of stack-level for distilled student models
@@ -149,6 +155,7 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         save_data: bool = False,
         random_state: int = 0,
         verbosity: int = 2,
+        raise_on_model_failure: bool = False,
     ):
         super().__init__(
             path=path,
@@ -163,6 +170,7 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         #: Integer value added to the stack level to get the random_state for kfold splits or the train/val split if bagging is disabled
         self.random_state = random_state
         self.verbosity = verbosity
+        self.raise_on_model_failure = raise_on_model_failure
 
         # TODO: consider redesign where Trainer doesn't need sample_weight column name and weights are separate from X
         self.sample_weight = sample_weight
@@ -2188,6 +2196,10 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
             # TODO: Add recursive=True to avoid repeatedly loading models each time this is called for bagged ensembles (especially during repeated bagging)
             self.save_model(model=model)
         except Exception as exc:
+            if self.raise_on_model_failure:
+                # immediately raise instead of skipping to next model, useful for debugging during development
+                logger.warning("Model failure occurred... Raising exception instead of continuing to next model. (raise_on_model_failure=True)")
+                raise exc
             exception = exc  # required to reference exc outside of `except` statement
             del_model = True
             if isinstance(exception, TimeLimitExceeded):

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.2.1b20250226"
+__version__ = "1.2.1b20250227"
 __lite__ = False

{autogluon.tabular-1.2.1b20250226.dist-info → autogluon.tabular-1.2.1b20250227.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250226
+Version: 1.2.1b20250227
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,19 +41,19 @@ Requires-Dist: scipy<1.16,>=1.5.4
 Requires-Dist: pandas<2.3.0,>=2.0.0
 Requires-Dist: scikit-learn<1.5.3,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.2.1b20250226
-Requires-Dist: autogluon.features==1.2.1b20250226
+Requires-Dist: autogluon.core==1.2.1b20250227
+Requires-Dist: autogluon.features==1.2.1b20250227
 Provides-Extra: all
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "all"
-Requires-Dist: lightgbm<4.6,>=4.0; extra == "all"
-Requires-Dist: autogluon.core[all]==1.2.1b20250226; extra == "all"
 Requires-Dist: fastai<2.8,>=2.3.1; extra == "all"
 Requires-Dist: xgboost<2.2,>=1.6; extra == "all"
-Requires-Dist: spacy<3.8; extra == "all"
-Requires-Dist: torch<2.6,>=2.2; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: lightgbm<4.6,>=4.0; extra == "all"
 Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: torch<2.6,>=2.2; extra == "all"
+Requires-Dist: spacy<3.8; extra == "all"
+Requires-Dist: autogluon.core[all]==1.2.1b20250227; extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -66,7 +66,7 @@ Requires-Dist: imodels<1.4.0,>=1.3.10; extra == "imodels"
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.6,>=4.0; extra == "lightgbm"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.2.1b20250226; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.2.1b20250227; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.1,>=2024.0; extra == "skex"
 Provides-Extra: skl2onnx

{autogluon.tabular-1.2.1b20250226.dist-info → autogluon.tabular-1.2.1b20250227.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.2.1b20250226-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.2.1b20250227-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=ADP5SToVDuyj8YndEeaK4OQXYSKGWWvpOifXJZvQoLU,91
+autogluon/tabular/version.py,sha256=5ozpv_b_rM463xTXRdXij1xDVDTJr5oIBlgIcEiF2lw,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/config_helper.py,sha256=Pb2aW9Z9w77pYKPRVZ3nBzHY3KJaiEJSJ747zZcJIVk,21132
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
@@ -14,8 +14,8 @@ autogluon/tabular/experimental/_tabular_classifier.py,sha256=7lGoFdvkHiZS3VpcXo9
 autogluon/tabular/experimental/_tabular_regressor.py,sha256=EzEDL-19T5QUVNmLkSHNzzGwYrUxyqlNpIDPMgtV6Gg,1932
 autogluon/tabular/experimental/plot_leaderboard.py,sha256=BN_kB-zmOZNUYWyI7z9pF67GCV20zo8yV51HKKj1SCY,9481
 autogluon/tabular/learner/__init__.py,sha256=Hhmk5WpKQHohVmI-veOaKMelKJpIdzeXrmw_DPn3DTU,63
-autogluon/tabular/learner/abstract_learner.py,sha256=3myDh867x-EWTPR-O-iw82WGgd5n1NKWf3kaTcYQeh0,54955
-autogluon/tabular/learner/default_learner.py,sha256=cg3K0oA-4ccXWtmGgy6qUJaLldieFwDvnMP_PyE9gdk,24579
+autogluon/tabular/learner/abstract_learner.py,sha256=HmnW7KO3sV4H1QSquJn9DYOwoWa2vohKThq_hr4OHM4,55102
+autogluon/tabular/learner/default_learner.py,sha256=hjdKbcFtIQxQ3-k1LiGOo-w5sLxIIQAyFLs3-R35aw0,24781
 autogluon/tabular/models/__init__.py,sha256=tDVqwVG9q2ctLWRouyXeYs5NiSBnOnwh3anAfZyM3jg,1099
 autogluon/tabular/models/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/_utils/rapids_utils.py,sha256=gbej9Hjn4alCWZuGN9sOLXMMAyWbgHPThTsp2feS39o,1038
@@ -144,23 +144,23 @@ autogluon/tabular/models/xt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 autogluon/tabular/models/xt/xt_model.py,sha256=z9Hcp3CR_QS4XMlE9-g5rFknj2KPHfFLtyqb0qvOYbw,825
 autogluon/tabular/predictor/__init__.py,sha256=zCMgjxQlWpDWnr1l1xjBCiK3rWC3N3RoD8UXBnazT74,107
 autogluon/tabular/predictor/interpretable_predictor.py,sha256=5UeKgnMFsfY65tiO3kxfHBPr03lyswLrgdtjPhI0Y7Q,6934
-autogluon/tabular/predictor/predictor.py,sha256=MSJjTqlfGEN4ESpUFCjdrnCbzv8-6pf89uiun2OSZTg,356361
+autogluon/tabular/predictor/predictor.py,sha256=A86C_9ixByrSrcU-Bxy9eTTZ7nCnIvRDN2uY-0ANLVY,357082
 autogluon/tabular/register/__init__.py,sha256=7CLOTWIUho0wi4eAwhYJ5Y0PfvNCWKnRwlw3bwYoTNE,93
 autogluon/tabular/register/_ag_model_register.py,sha256=tMr-QgxgCE49tdThdSFOZaJg2D9ckDh6fiR5K4cRtvk,1564
 autogluon/tabular/register/_model_register.py,sha256=jqSg0d89dXAAcp-OT4II90ce994ByKMMzAYmpkyaRbI,6824
 autogluon/tabular/trainer/__init__.py,sha256=PW_PGL-tWoQzx3ES2S53bQEZOtsRWTYiM9QdOqsk0dI,38
-autogluon/tabular/trainer/abstract_trainer.py,sha256=daz6_IfYX9bvQ5c5RpbxZ9f5JytUDq2-XDO1wAXpXtk,231155
+autogluon/tabular/trainer/abstract_trainer.py,sha256=lqOjVTLUaZNue4B7u47PYXTXsBEFbSJ4SyruNeChFCk,231925
 autogluon/tabular/trainer/auto_trainer.py,sha256=FyRWM8iUJuDvw_aqV5EV_xdh_pb-nHzAvG1sbEhvs0g,8680
 autogluon/tabular/trainer/model_presets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/trainer/model_presets/presets.py,sha256=bTPGPyz07a7GG6327yO6ryuWbNc1aq3hF1qzZL-Xe4c,16733
 autogluon/tabular/trainer/model_presets/presets_distill.py,sha256=MnFC2GJc6RmDBNAGbsO2XMfo3PjR8cUrZoilWW8gTYQ,3295
 autogluon/tabular/tuning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/tuning/feature_pruner.py,sha256=9iNku8gVbYEkjuKlyITPJDicsNkoraaQOlINQq9iZlQ,6877
-autogluon.tabular-1.2.1b20250226.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.2.1b20250226.dist-info/METADATA,sha256=w27z14Cxm2sznG8jKz1q11QwLJOcc_LBsz0VbrXzLek,14386
-autogluon.tabular-1.2.1b20250226.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.2.1b20250226.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.2.1b20250226.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250226.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250226.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.2.1b20250226.dist-info/RECORD,,
+autogluon.tabular-1.2.1b20250227.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.2.1b20250227.dist-info/METADATA,sha256=nw0R8W2F-rUL_zr6eevigQGHEmmFKBhH2TGfPHvjuUE,14386
+autogluon.tabular-1.2.1b20250227.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.2.1b20250227.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.2.1b20250227.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250227.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250227.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.2.1b20250227.dist-info/RECORD,,