autogluon.core 1.2.1b20250306__py3-none-any.whl → 1.2.1b20250308__py3-none-any.whl

autogluon/core/models/ensemble/bagged_ensemble_model.py

@@ -620,7 +620,8 @@ class BaggedEnsembleModel(AbstractModel):
             # FIXME: Consider use_child_oof with pseudo labels! Need to keep track of indices
             logger.log(15, f"{len(X_pseudo)} extra rows of pseudolabeled data added to training set for {self.name}")
             assert_pseudo_column_match(X=X, X_pseudo=X_pseudo)
-            X_fit = pd.concat([X, X_pseudo], axis=0, ignore_index=True)
+            # Needs .astype(X.dtypes) because pd.concat will convert categorical features to int/float unexpectedly. Need to convert them back to original.
+            X_fit = pd.concat([X, X_pseudo], axis=0, ignore_index=True).astype(X.dtypes)
             y_fit = pd.concat([y, y_pseudo], axis=0, ignore_index=True)
         else:
             X_fit = X

autogluon/core/version.py

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

{autogluon.core-1.2.1b20250306.dist-info → autogluon.core-1.2.1b20250308.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-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
@@ -45,11 +45,11 @@ Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: requests
 Requires-Dist: matplotlib<3.11,>=3.7.0
 Requires-Dist: boto3<2,>=1.10
-Requires-Dist: autogluon.common==1.2.1b20250306
+Requires-Dist: autogluon.common==1.2.1b20250308
 Provides-Extra: all
-Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
-Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "all"
 Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "all"
+Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "all"
+Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
 Requires-Dist: pyarrow>=15.0.0; extra == "all"
 Provides-Extra: ray
 Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "ray"
@@ -58,11 +58,11 @@ Requires-Dist: pyarrow>=15.0.0; extra == "raytune"
 Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "raytune"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "raytune"
 Provides-Extra: tests
+Requires-Dist: pytest-mypy; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
-Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: types-requests; extra == "tests"
 Requires-Dist: flake8; extra == "tests"
-Requires-Dist: pytest-mypy; extra == "tests"
+Requires-Dist: types-setuptools; extra == "tests"
 
 
 

{autogluon.core-1.2.1b20250306.dist-info → autogluon.core-1.2.1b20250308.dist-info}/RECORD

@@ -1,9 +1,9 @@
-autogluon.core-1.2.1b20250306-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.core-1.2.1b20250308-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
 autogluon/core/_setup_utils.py,sha256=ikxn4zc3PNyjJJT5SsgL0dvP-6Rbq6_dItGMiZNINv4,6958
 autogluon/core/constants.py,sha256=nEVLdSFJ-5O-tz3jUD3qPX65RMp7g8qOR38XlurbP4Y,3403
 autogluon/core/problem_type.py,sha256=XJmMgeNBgS7u43pDK-spTivatPyh_INOXveEXwQt-Rw,2993
-autogluon/core/version.py,sha256=vpLPQw8Tz1FiEFDl7STt_ZrtxXTVYzhhywUGcg_8Fec,91
+autogluon/core/version.py,sha256=hEltA28xjQZluR9kZu7xMHB0PNrfraZqGe-bAnGshFE,91
 autogluon/core/augmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/augmentation/distill_utils.py,sha256=JBlp2WOMNKoJv8aKVwJVRQSalSk8jx36HM7-k_VvkhY,9404
 autogluon/core/calibrate/__init__.py,sha256=eU6qLj7DKUhaz2HHNHDrfroRaLM-mhuSncK_v1UP4F8,62
@@ -47,7 +47,7 @@ autogluon/core/models/dummy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 autogluon/core/models/dummy/_dummy_quantile_regressor.py,sha256=i-ZW2flJ60jsMfMK24IP39Xwc55-UlBDvHmqanIf29Q,664
 autogluon/core/models/dummy/dummy_model.py,sha256=oCAL0HnPJEREPtgnS_N8YtCdn99yXW0zD8EBEJJToNA,1635
 autogluon/core/models/ensemble/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/core/models/ensemble/bagged_ensemble_model.py,sha256=kQbJhq_UzUoOHsTX00WTEXu1Ogdme4bdLiiWhIMNUfU,77257
+autogluon/core/models/ensemble/bagged_ensemble_model.py,sha256=bR-T5mnZZRgRF_XRar_5B9vAdGR9-V6KShg3xq76xXg,77430
 autogluon/core/models/ensemble/fold_fitting_strategy.py,sha256=01vzNVvE4FIFgD6YqbhK63XoUlSztnVFsrDdsoqm75U,47021
 autogluon/core/models/ensemble/ray_parallel_fold_fitting_strategy.py,sha256=8RASa-eV6n9kUgbqQHNt7k4IrvuB9NdrunIMLYOLwgA,2068
 autogluon/core/models/ensemble/stacker_ensemble_model.py,sha256=oGgn0oxu1jzygcCHKBvlMZLbKrDAIgcrOMXFbSTNTVM,18059
@@ -89,11 +89,11 @@ autogluon/core/utils/utils.py,sha256=FMa9kIUAxA3IIBbATmBnNEVObSAivehZ2_zCy3PRR-c
 autogluon/core/utils/version_utils.py,sha256=5-r8hLRKTaZbj5qo2uzE_2E4casH49Ye3WyeHlgHuz4,3252
 autogluon/core/utils/loaders/__init__.py,sha256=W5FAdQvpDcn_uisqJrlSAObWVta-YjJLKGN3NCbEgIo,109
 autogluon/core/utils/savers/__init__.py,sha256=bGWciSxAkj6u06vOC4pTvr22f_1ey0glgvmjCMEOm78,89
-autogluon.core-1.2.1b20250306.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.core-1.2.1b20250306.dist-info/METADATA,sha256=zu01FR8M6niA3jS9iV6e40t2WAi0VJPAxm8JY53nsBU,12399
-autogluon.core-1.2.1b20250306.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.core-1.2.1b20250306.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.core-1.2.1b20250306.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250306.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250306.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.core-1.2.1b20250306.dist-info/RECORD,,
+autogluon.core-1.2.1b20250308.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.core-1.2.1b20250308.dist-info/METADATA,sha256=wbyqy6pdn10wA-EbKov-yb3Z-HRjxXruJQ9krNUGtlQ,12399
+autogluon.core-1.2.1b20250308.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.core-1.2.1b20250308.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.core-1.2.1b20250308.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250308.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250308.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.core-1.2.1b20250308.dist-info/RECORD,,