autogluon.tabular 1.2.1b20250306__tar.gz → 1.2.1b20250308__tar.gz

{autogluon.tabular-1.2.1b20250306 → autogluon.tabular-1.2.1b20250308}/PKG-INFO

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

{autogluon.tabular-1.2.1b20250306 → autogluon.tabular-1.2.1b20250308}/setup.py

@@ -34,7 +34,7 @@ install_requires = [
 
 extras_require = {
     "lightgbm": [
-        "lightgbm>=4.0,<4.6",  # <{N+1} upper cap, where N is the latest released minor version
+        "lightgbm>=4.0,<4.7",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "catboost": [
         "numpy>=1.25,<2.0.0",  # TODO support numpy>=2.0.0 once issue resolved https://github.com/catboost/catboost/issues/2671
@@ -71,14 +71,13 @@ extras_require = {
         # No vowpalwabbit wheel for python 3.11 or above yet
         "vowpalwabbit>=9,<9.10; python_version < '3.11' and sys_platform != 'darwin'",
     ],
-    
 }
 
-is_aarch64 = platform.machine() == 'aarch64'
-is_darwin = sys.platform == 'darwin'
+is_aarch64 = platform.machine() == "aarch64"
+is_darwin = sys.platform == "darwin"
 
 if is_darwin or is_aarch64:
-# For macOS or aarch64, only use CPU version
+    # For macOS or aarch64, only use CPU version
     extras_require["skl2onnx"] = [
         "onnx>=1.13.0,<1.16.2;platform_system=='Windows'",  # cap at 1.16.1 for issue https://github.com/onnx/onnx/issues/6267
         "onnx>=1.13.0,<1.18.0;platform_system!='Windows'",
@@ -88,12 +87,12 @@ if is_darwin or is_aarch64:
         "onnxruntime>=1.17.0,<1.20.0",
     ]
 else:
-# For other platforms, include both CPU and GPU versions
+    # For other platforms, include both CPU and GPU versions
     extras_require["skl2onnx"] = [
         "onnx>=1.13.0,<1.16.2;platform_system=='Windows'",  # cap at 1.16.1 for issue https://github.com/onnx/onnx/issues/6267
         "onnx>=1.13.0,<1.18.0;platform_system!='Windows'",
-        "skl2onnx>=1.15.0,<1.18.0", 
-        "onnxruntime>=1.17.0,<1.20.0",   # install for gpu system due to https://github.com/autogluon/autogluon/issues/3804
+        "skl2onnx>=1.15.0,<1.18.0",
+        "onnxruntime>=1.17.0,<1.20.0",  # install for gpu system due to https://github.com/autogluon/autogluon/issues/3804
         "onnxruntime-gpu>=1.17.0,<1.20.0",
     ]
 

{autogluon.tabular-1.2.1b20250306 → autogluon.tabular-1.2.1b20250308}/src/autogluon/tabular/trainer/abstract_trainer.py

@@ -2148,8 +2148,9 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         # is required
         if not isinstance(model, BaggedEnsembleModel) and X_pseudo is not None and y_pseudo is not None and X_pseudo.columns.equals(X.columns):
             assert_pseudo_column_match(X=X, X_pseudo=X_pseudo)
-            X_w_pseudo = pd.concat([X, X_pseudo])
-            y_w_pseudo = pd.concat([y, y_pseudo])
+            # Needs .astype(X.dtypes) because pd.concat will convert categorical features to int/float unexpectedly. Need to convert them back to original.
+            X_w_pseudo = pd.concat([X, X_pseudo], ignore_index=True).astype(X.dtypes)
+            y_w_pseudo = pd.concat([y, y_pseudo], ignore_index=True)
             logger.log(15, f"{len(X_pseudo)} extra rows of pseudolabeled data added to training set for {model.name}")
             model_fit_kwargs["X"] = X_w_pseudo
             model_fit_kwargs["y"] = y_w_pseudo

{autogluon.tabular-1.2.1b20250306 → autogluon.tabular-1.2.1b20250308}/src/autogluon/tabular/version.py

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

{autogluon.tabular-1.2.1b20250306 → autogluon.tabular-1.2.1b20250308}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.2.1b20250306 → autogluon.tabular-1.2.1b20250308}/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.1b20250306
-autogluon.features==1.2.1b20250306
+autogluon.core==1.2.1b20250308
+autogluon.features==1.2.1b20250308
 
 [all]
 spacy<3.8
-catboost<1.3,>=1.2
+lightgbm<4.7,>=4.0
+xgboost<2.2,>=2.0
+autogluon.core[all]==1.2.1b20250308
+huggingface_hub[torch]
 torch<2.6,>=2.2
 fastai<2.8,>=2.3.1
-huggingface_hub[torch]
-lightgbm<4.6,>=4.0
-autogluon.core[all]==1.2.1b20250306
-einops<0.9,>=0.7
+catboost<1.3,>=1.2
 numpy<2.0.0,>=1.25
-xgboost<2.2,>=2.0
+einops<0.9,>=0.7
 
 [catboost]
 numpy<2.0.0,>=1.25
@@ -31,10 +31,10 @@ fastai<2.8,>=2.3.1
 imodels<1.4.0,>=1.3.10
 
 [lightgbm]
-lightgbm<4.6,>=4.0
+lightgbm<4.7,>=4.0
 
 [ray]
-autogluon.core[all]==1.2.1b20250306
+autogluon.core[all]==1.2.1b20250308
 
 [skex]
 scikit-learn-intelex<2025.1,>=2024.0