autogluon.tabular 1.2.1b20250415__py3-none-any.whl → 1.2.1b20250417__py3-none-any.whl

autogluon/tabular/trainer/abstract_trainer.py

@@ -2322,9 +2322,16 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         boolean, True if model was registered, False if model was found to be invalid and not registered.
         """
         if model.val_score is not None and np.isnan(model.val_score):
-            logger.warning(
-                f"WARNING: {model.name} has a val_score of {model.val_score} (NaN)! This should never happen. The model will not be saved to avoid instability."
+            msg = (
+                f"WARNING: {model.name} has a val_score of {model.val_score} (NaN)! "
+                f"This should never happen. The model will not be saved to avoid instability."
             )
+            if self.raise_on_model_failure:
+                raise AssertionError(
+                    f"{msg} Raising an exception because `raise_on_model_failure={self.raise_on_model_failure}`."
+                )
+            else:
+                logger.warning(msg)
             return False
         # TODO: Add to HPO
 

autogluon/tabular/version.py

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

{autogluon.tabular-1.2.1b20250415.dist-info → autogluon.tabular-1.2.1b20250417.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250415
+Version: 1.2.1b20250417
 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.7.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.2.1b20250415
-Requires-Dist: autogluon.features==1.2.1b20250415
+Requires-Dist: autogluon.core==1.2.1b20250417
+Requires-Dist: autogluon.features==1.2.1b20250417
 Provides-Extra: all
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
-Requires-Dist: catboost<1.3,>=1.2; extra == "all"
-Requires-Dist: spacy<3.8; extra == "all"
+Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
-Requires-Dist: huggingface-hub[torch]; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: autogluon.core[all]==1.2.1b20250415; extra == "all"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
 Requires-Dist: torch<2.7,>=2.2; extra == "all"
+Requires-Dist: autogluon.core[all]==1.2.1b20250417; extra == "all"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
+Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: spacy<3.8; 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<2.1.0,>=1.3.10; extra == "imodels"
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "lightgbm"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.2.1b20250415; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.2.1b20250417; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.5,>=2024.0; extra == "skex"
 Provides-Extra: skl2onnx

{autogluon.tabular-1.2.1b20250415.dist-info → autogluon.tabular-1.2.1b20250417.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.2.1b20250415-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.2.1b20250417-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=b5iA_UuQQbMm77xD8yblOR_jEE8jGtdLXDEMT2oS0Lk,91
+autogluon/tabular/version.py,sha256=9DNociU3CkjmMks3rFL-2UfsFfw26T7_cq6VAom-ic4,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
@@ -139,18 +139,18 @@ autogluon/tabular/testing/fit_helper.py,sha256=gVHTdAsp_lSZ_qbwjXM7aA5fI32zHj3_z
 autogluon/tabular/testing/generate_datasets.py,sha256=UXPNfviUNZqGcx4mTYIloJxRED6DRMxAgHqbOvjEUCs,3603
 autogluon/tabular/testing/model_fit_helper.py,sha256=ZjWpw2nyeFnsrccmkfQtx3qbA8HJx282XX2rwdS-LIs,3808
 autogluon/tabular/trainer/__init__.py,sha256=PW_PGL-tWoQzx3ES2S53bQEZOtsRWTYiM9QdOqsk0dI,38
-autogluon/tabular/trainer/abstract_trainer.py,sha256=EEjZCcO5W-034o_fzbmcIYbJD7xt1K7pYiw3zPZEhUk,232136
+autogluon/tabular/trainer/abstract_trainer.py,sha256=F9bT78KHyQi96FbPs6ufaswVQJBzxoK3sTXyRXXpK_g,232416
 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.1b20250415.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.2.1b20250415.dist-info/METADATA,sha256=mHmkIllgm_ZuWiRKAgT_ONufsSEo3mYbwpooHQo5Yjg,14069
-autogluon.tabular-1.2.1b20250415.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.2.1b20250415.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.2.1b20250415.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250415.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250415.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.2.1b20250415.dist-info/RECORD,,
+autogluon.tabular-1.2.1b20250417.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.2.1b20250417.dist-info/METADATA,sha256=KtK6Wl-kEnQ2kNajgxKGz_lQUDPD6hZqlef4klkqsrM,14069
+autogluon.tabular-1.2.1b20250417.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.2.1b20250417.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.2.1b20250417.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250417.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250417.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.2.1b20250417.dist-info/RECORD,,