autogluon.tabular 1.2.1b20250112__tar.gz → 1.2.1b20250114__tar.gz

{autogluon.tabular-1.2.1b20250112 → autogluon.tabular-1.2.1b20250114}/PKG-INFO

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

{autogluon.tabular-1.2.1b20250112 → autogluon.tabular-1.2.1b20250114}/setup.py

@@ -52,7 +52,8 @@ extras_require = {
         "fastai>=2.3.1,<2.8",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "tabpfn": [
-        "tabpfn>=0.1,<0.2",  # <{N+1} upper cap, where N is the latest released minor version
+        # versions below 0.1.11 are broken
+        "tabpfn>=0.1.11,<0.2",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "tabpfnmix": [
         "torch",  # version range defined in `core/_setup_utils.py`

{autogluon.tabular-1.2.1b20250112 → autogluon.tabular-1.2.1b20250114}/src/autogluon/tabular/predictor/predictor.py

@@ -1388,10 +1388,19 @@ class TabularPredictor(TabularPredictorDeprecatedMixin):
             )
         else:
             # Holdout is false, use (repeated) cross-validation
-            is_stratified = self.problem_type in [REGRESSION, QUANTILE, SOFTCLASS]
+            is_stratified = self.problem_type in [BINARY, MULTICLASS]
+            is_binned = self.problem_type in [REGRESSION, QUANTILE]
             self._learner._validate_groups(X=X, X_val=X_val)  # Validate splits before splitting
-            splits = CVSplitter(n_splits=n_folds, n_repeats=n_repeats, groups=self._learner.groups, stratified=is_stratified, random_state=42).split(
-                X=X.drop(self.label, axis=1), y=X[self.label]
+            splits = CVSplitter(
+                n_splits=n_folds,
+                n_repeats=n_repeats,
+                groups=self._learner.groups,
+                stratify=is_stratified,
+                bin=is_binned,
+                random_state=42,
+            ).split(
+                X=X.drop(self.label, axis=1),
+                y=X[self.label]
             )
             n_splits = len(splits)
             logger.info(

{autogluon.tabular-1.2.1b20250112 → autogluon.tabular-1.2.1b20250114}/src/autogluon/tabular/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.2.1b20250112'
+__version__ = '1.2.1b20250114'
 __lite__ = False

{autogluon.tabular-1.2.1b20250112 → autogluon.tabular-1.2.1b20250114}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.2.1b20250112 → autogluon.tabular-1.2.1b20250114}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,20 +3,20 @@ scipy<1.16,>=1.5.4
 pandas<2.3.0,>=2.0.0
 scikit-learn<1.5.3,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.2.1b20250112
-autogluon.features==1.2.1b20250112
+autogluon.core==1.2.1b20250114
+autogluon.features==1.2.1b20250114
 
 [all]
-catboost<1.3,>=1.2
-xgboost<2.2,>=1.6
-torch<2.6,>=2.2
-autogluon.core[all]==1.2.1b20250112
 numpy<2.0.0,>=1.25
-spacy<3.8
-lightgbm<4.6,>=4.0
 fastai<2.8,>=2.3.1
-huggingface_hub[torch]
+lightgbm<4.6,>=4.0
+torch<2.6,>=2.2
+spacy<3.8
+xgboost<2.2,>=1.6
+autogluon.core[all]==1.2.1b20250114
 einops<0.9,>=0.7
+catboost<1.3,>=1.2
+huggingface_hub[torch]
 
 [catboost]
 numpy<2.0.0,>=1.25
@@ -34,7 +34,7 @@ imodels<1.4.0,>=1.3.10
 lightgbm<4.6,>=4.0
 
 [ray]
-autogluon.core[all]==1.2.1b20250112
+autogluon.core[all]==1.2.1b20250114
 
 [skex]
 scikit-learn-intelex<2025.1,>=2024.0
@@ -51,7 +51,7 @@ onnx<1.18.0,>=1.13.0
 onnx<1.16.2,>=1.13.0
 
 [tabpfn]
-tabpfn<0.2,>=0.1
+tabpfn<0.2,>=0.1.11
 
 [tabpfnmix]
 torch<2.6,>=2.2
@@ -59,7 +59,7 @@ huggingface_hub[torch]
 einops<0.9,>=0.7
 
 [tests]
-tabpfn<0.2,>=0.1
+tabpfn<0.2,>=0.1.11
 torch<2.6,>=2.2
 huggingface_hub[torch]
 einops<0.9,>=0.7