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

autogluon/tabular/models/fastainn/tabular_nn_fastai.py

@@ -41,6 +41,8 @@ from autogluon.tabular.models.tabular_nn.utils.nn_architecture_utils import infe
 from .hyperparameters.parameters import get_param_baseline
 from .hyperparameters.searchspaces import get_default_searchspace
 
+warnings.filterwarnings("ignore", message="load_learner` uses Python's insecure pickle module")
+
 # FIXME: Has a leak somewhere, training additional models in a single python script will slow down training for each additional model. Gets very slow after 20+ models (10x+ slowdown)
 #  Slowdown does not appear to impact Mac OS
 # Reproduced with raw torch: https://github.com/pytorch/pytorch/issues/31867
@@ -536,6 +538,7 @@ class NNFastAiTabularModel(AbstractModel):
 
     @classmethod
     def load(cls, path: str, reset_paths=True, verbose=True):
+
         from fastai.learner import load_learner
 
         model = super().load(path, reset_paths=reset_paths, verbose=verbose)

autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_classifier.py

@@ -64,4 +64,4 @@ class TabPFNMixClassifier(BaseEstimator, ClassifierMixin):
     # FIXME: Avoid preprocessing self.X_ and self.y_ each predict_proba call
     def predict_proba(self, X):
         logits = self.trainer.predict(self.X_, self.y_, X)
-        return np.exp(logits) / np.exp(logits).sum(axis=1)[:, None]
+        return np.exp(logits) / np.exp(logits).sum(axis=1)[:, None]

autogluon/tabular/models/tabular_nn/utils/categorical_encoders.py

@@ -6,9 +6,11 @@ Unknown categories are returned as None in inverse transforms. Always converts i
 
 import copy
 from numbers import Integral
+from packaging.version import parse as parse_version
 
 import numpy as np
 from scipy import sparse
+from sklearn import __version__ as _sklearn_version
 from sklearn.base import BaseEstimator, TransformerMixin
 from sklearn.utils import check_array
 from sklearn.utils.validation import check_is_fitted
@@ -162,7 +164,10 @@ class _BaseEncoder(BaseEstimator, TransformerMixin):
         """
         if not (hasattr(X, "iloc") and getattr(X, "ndim", 0) == 2):
             # if not a dataframe, do normal check_array validation
-            X_temp = check_array(X, dtype=None, force_all_finite=False)
+            if parse_version(_sklearn_version) >= parse_version("1.6.0"):
+                X_temp = check_array(X, dtype=None, ensure_all_finite=False)
+            else:
+                X_temp = check_array(X, dtype=None, force_all_finite=False)
             if not hasattr(X, "dtype") and np.issubdtype(X_temp.dtype, np.str_):
                 X = check_array(X, dtype=object)
             else:
@@ -178,7 +183,10 @@ class _BaseEncoder(BaseEstimator, TransformerMixin):
 
         for i in range(n_features):
             Xi = self._get_feature(X, feature_idx=i)
-            Xi = check_array(Xi, ensure_2d=False, dtype=None, force_all_finite=needs_validation)
+            if parse_version(_sklearn_version) >= parse_version("1.6.0"):
+                Xi = check_array(Xi, ensure_2d=False, dtype=None, ensure_all_finite=needs_validation)
+            else:
+                Xi = check_array(Xi, ensure_2d=False, dtype=None, force_all_finite=needs_validation)
             X_columns.append(Xi)
 
         return X_columns, n_samples, n_features
@@ -304,6 +312,39 @@ class _BaseEncoder(BaseEstimator, TransformerMixin):
     def _more_tags(self):
         return {"X_types": ["categorical"]}
 
+    def __sklearn_tags__(self):
+        """
+        Returns a Tags object with scikit-learn estimator tags.
+
+        This is the scikit-learn 1.6+ compatible way to define estimator tags,
+        replacing the deprecated _more_tags method.
+
+        Returns
+        -------
+        tags : sklearn.utils.Tags
+            A Tags object containing all tag information.
+        """
+        # lazily import to avoid crashing if sklearn<1.6
+        from sklearn.utils import Tags, InputTags, TargetTags
+
+        # Create the Tags object with appropriate settings
+        tags = Tags(
+            estimator_type=None,  # This is a transformer, not a classifier/regressor
+            target_tags=TargetTags(
+                required=False  # Target is not required for transformers
+            ),
+            input_tags=InputTags(
+                categorical=True,
+                string=True,
+            ),
+            array_api_support=False,
+            no_validation=False,
+            non_deterministic=False,
+            requires_fit=True,
+        )
+
+        return tags
+
 
 class OneHotMergeRaresHandleUnknownEncoder(_BaseEncoder):
     """Encode categorical integer features as a one-hot numeric array.

autogluon/tabular/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250413
+Version: 1.2.1b20250415
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -39,21 +39,21 @@ License-File: ../NOTICE
 Requires-Dist: numpy<2.3.0,>=1.25.0
 Requires-Dist: scipy<1.16,>=1.5.4
 Requires-Dist: pandas<2.3.0,>=2.0.0
-Requires-Dist: scikit-learn<1.6.0,>=1.4.0
+Requires-Dist: scikit-learn<1.7.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.2.1b20250413
-Requires-Dist: autogluon.features==1.2.1b20250413
+Requires-Dist: autogluon.core==1.2.1b20250415
+Requires-Dist: autogluon.features==1.2.1b20250415
 Provides-Extra: all
-Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: numpy<2.0.0,>=1.25; extra == "all"
-Requires-Dist: torch<2.7,>=2.2; extra == "all"
-Requires-Dist: spacy<3.8; extra == "all"
-Requires-Dist: autogluon.core[all]==1.2.1b20250413; extra == "all"
-Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; 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: 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"
 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.1b20250413; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.2.1b20250415; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.5,>=2024.0; extra == "skex"
 Provides-Extra: skl2onnx

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

@@ -1,6 +1,6 @@
-autogluon.tabular-1.2.1b20250413-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.2.1b20250415-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=-0zBgl90kQhkR8VGoVQR1GnZlG3Y5dZpwyqECIWshWc,91
+autogluon/tabular/version.py,sha256=b5iA_UuQQbMm77xD8yblOR_jEE8jGtdLXDEMT2oS0Lk,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
@@ -36,7 +36,7 @@ autogluon/tabular/models/fastainn/callbacks.py,sha256=3WvOEwqd1YAVInooKsFOTzAkCL
 autogluon/tabular/models/fastainn/fastai_helpers.py,sha256=gGYzyrAFl8hi8GnsemZNLGZn5xr7cyJXdFl08PIlza4,1393
 autogluon/tabular/models/fastainn/imports_helper.py,sha256=ICxA8ty47-oZu0Q9AjKCQe8uVi340Iu0NFruxvJPrbA,330
 autogluon/tabular/models/fastainn/quantile_helpers.py,sha256=d89GKvSRBgOy9EqcDI83MK5sqPRxP6JJ3BmPLmKnB0o,1808
-autogluon/tabular/models/fastainn/tabular_nn_fastai.py,sha256=OqiOP3dw9HsZCw9AXvJqlTNwheupkuhaMiKWDnVLpjc,29424
+autogluon/tabular/models/fastainn/tabular_nn_fastai.py,sha256=wtvs2VclaEvt-DDtTAxteVpDTXTAYPBp72jGxlGQwgE,29522
 autogluon/tabular/models/fastainn/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/fastainn/hyperparameters/parameters.py,sha256=DkQwAZZ7CuODKoljr-yrkx-uFxBSPRxkKuvPdwO-UhQ,2069
 autogluon/tabular/models/fastainn/hyperparameters/searchspaces.py,sha256=5qdknZDrHtdPdrhSqjamYQrCxvupXvlN3bVGEPgs48E,1660
@@ -79,7 +79,7 @@ autogluon/tabular/models/tabpfn/tabpfn_model.py,sha256=PEYMuIh5TFLIDy3hcjfz1DcvD
 autogluon/tabular/models/tabpfnmix/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabpfnmix/tabpfnmix_model.py,sha256=Bo-JMnNgI0fYXXTHy3zLAE1OHZv9ikgH4bFBaVSa79g,16174
 autogluon/tabular/models/tabpfnmix/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_classifier.py,sha256=U3DAhtSLpHUu-22GgR3QCZJnpRYpOW95XoTV7tE9J5Y,3425
+autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_classifier.py,sha256=_WIO_YQBUCfprKYLHxUNEICPb5XWZw4zbw00DuiTk_s,3426
 autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_regressor.py,sha256=J6JvrK6L6y3s-Ah6sHQdjSK0mwAMP-Wy3RRBwzB0AoA,3196
 autogluon/tabular/models/tabpfnmix/_internal/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabpfnmix/_internal/config/config_run.py,sha256=dnyEBOIS3QX4_JsjepLMxsK8Qv-CTsE1gEIG-0v1YCU,232
@@ -114,7 +114,7 @@ autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py,sha256=tDxWg4SLUjB
 autogluon/tabular/models/tabular_nn/torch/tabular_torch_dataset.py,sha256=RdnQGZSrvY1iuJB4JTANniH3Dorw-DP0Em_JK3_h7RM,13497
 autogluon/tabular/models/tabular_nn/torch/torch_network_modules.py,sha256=Qc3PwXTD8A7PgXi6EGuaBCrN3jsFAXDLCW7i6tE5wYI,11338
 autogluon/tabular/models/tabular_nn/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabular_nn/utils/categorical_encoders.py,sha256=uLQaHkudg8AS2jFwNv4FyQ9KzO_F-dj6hIJe8R2t4aw,34213
+autogluon/tabular/models/tabular_nn/utils/categorical_encoders.py,sha256=2B5SrSN5nlCUGSsn2hrZNM5m4FswDKRxs_08CVB42js,35759
 autogluon/tabular/models/tabular_nn/utils/data_preprocessor.py,sha256=ypXqtxdt1qH6la1hcq-BJ0dzQBNtgKY-BjXmIWxPjCg,5237
 autogluon/tabular/models/tabular_nn/utils/nn_architecture_utils.py,sha256=tttzR5EtYcFa6sIrUG9wyegdYmYE5DPK_CiLF1-L3c8,2875
 autogluon/tabular/models/text_prediction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -146,11 +146,11 @@ autogluon/tabular/trainer/model_presets/presets.py,sha256=bTPGPyz07a7GG6327yO6ry
 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.1b20250413.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.2.1b20250413.dist-info/METADATA,sha256=F59f-WrfTSDKiRwWaA7clkfB0svB3iRK3Jw44H7GP9U,14069
-autogluon.tabular-1.2.1b20250413.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.2.1b20250413.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.2.1b20250413.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250413.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250413.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.2.1b20250413.dist-info/RECORD,,
+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,,