autogluon.tabular 1.2.1b20250414__tar.gz → 1.2.1b20250416__tar.gz

{autogluon.tabular-1.2.1b20250414 → autogluon.tabular-1.2.1b20250416}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250414
+Version: 1.2.1b20250416
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.tabular-1.2.1b20250414 → autogluon.tabular-1.2.1b20250416}/src/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-1.2.1b20250414 → autogluon.tabular-1.2.1b20250416}/src/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-1.2.1b20250414 → autogluon.tabular-1.2.1b20250416}/src/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-1.2.1b20250414 → autogluon.tabular-1.2.1b20250416}/src/autogluon/tabular/version.py

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

{autogluon.tabular-1.2.1b20250414 → autogluon.tabular-1.2.1b20250416}/src/autogluon.tabular.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250414
+Version: 1.2.1b20250416
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.tabular-1.2.1b20250414 → autogluon.tabular-1.2.1b20250416}/src/autogluon.tabular.egg-info/requires.txt

@@ -1,22 +1,22 @@
 numpy<2.3.0,>=1.25.0
 scipy<1.16,>=1.5.4
 pandas<2.3.0,>=2.0.0
-scikit-learn<1.6.0,>=1.4.0
+scikit-learn<1.7.0,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.2.1b20250414
-autogluon.features==1.2.1b20250414
+autogluon.core==1.2.1b20250416
+autogluon.features==1.2.1b20250416
 
 [all]
-einops<0.9,>=0.7
+spacy<3.8
 huggingface_hub[torch]
+autogluon.core[all]==1.2.1b20250416
+einops<0.9,>=0.7
+fastai<2.9,>=2.3.1
+numpy<2.0.0,>=1.25
+catboost<1.3,>=1.2
 torch<2.7,>=2.2
 xgboost<3.1,>=2.0
 lightgbm<4.7,>=4.0
-spacy<3.8
-catboost<1.3,>=1.2
-fastai<2.9,>=2.3.1
-numpy<2.0.0,>=1.25
-autogluon.core[all]==1.2.1b20250414
 
 [catboost]
 numpy<2.0.0,>=1.25
@@ -34,7 +34,7 @@ imodels<2.1.0,>=1.3.10
 lightgbm<4.7,>=4.0
 
 [ray]
-autogluon.core[all]==1.2.1b20250414
+autogluon.core[all]==1.2.1b20250416
 
 [skex]
 scikit-learn-intelex<2025.5,>=2024.0