autogluon.core 1.2.1b20250414__py3-none-any.whl → 1.2.1b20250416__py3-none-any.whl

autogluon/core/_setup_utils.py

@@ -19,7 +19,7 @@ DEPENDENT_PACKAGES = {
     "boto3": ">=1.10,<2",  # <2 because unlikely to introduce breaking changes in minor releases. >=1.10 because 1.10 is 3 years old, no need to support older
     "numpy": ">=1.25.0,<2.3.0",  # "<{N+3}" upper cap, where N is the latest released minor version, assuming no warnings using N
     "pandas": ">=2.0.0,<2.3.0",  # "<{N+3}" upper cap
-    "scikit-learn": ">=1.4.0,<1.6.0",  # For >=1.6.0 support, we need to use __sklearn_tags__ https://scikit-learn.org/stable/developers/develop.html#estimator-tags
+    "scikit-learn": ">=1.4.0,<1.7.0",  # <{N+1} upper cap
     "scipy": ">=1.5.4,<1.16",  # "<{N+2}" upper cap
     "matplotlib": ">=3.7.0,<3.11",  # "<{N+2}" upper cap
     "psutil": ">=5.7.3,<7.1.0",  # Major version cap
@@ -30,7 +30,7 @@ DEPENDENT_PACKAGES = {
     "torch": ">=2.2,<2.7",  # Major version cap, sync with common/src/autogluon/common/utils/try_import.py
     "lightning": ">=2.2,<2.7",  # Major version cap
     "async_timeout": ">=4.0,<6",  # Major version cap
-    "transformers[sentencepiece]": ">=4.38.0,<4.50", # there is a breaking change in 4.50 for model config saving
+    "transformers[sentencepiece]": ">=4.38.0,<4.50",  # there is a breaking change in 4.50 for model config saving
     "accelerate": ">=0.34.0,<2.0",
     "typing-extensions": ">=4.0,<5",
 }

autogluon/core/models/abstract/abstract_model.py

@@ -79,6 +79,7 @@ class Taggable(ABC):
                 collected_tags.update(more_tags)
         return collected_tags
 
+
     @classmethod
     def _get_class_tags(cls) -> dict:
         """
@@ -797,8 +798,8 @@ class AbstractModel(ModelBase, Tunable):
             ), f"Specified num_cpus per {self.__class__.__name__} is more than the total: {system_num_cpus}"
         if user_specified_lower_level_num_gpus is not None:
             assert (
-                user_specified_lower_level_num_gpus <= system_num_cpus
-            ), f"Specified num_gpus per {self.__class__.__name__} is more than the total: {system_num_cpus}"
+                user_specified_lower_level_num_gpus <= system_num_gpus
+            ), f"Specified num_gpus per {self.__class__.__name__} is more than the total: {system_num_gpus}"
         k_fold = kwargs.get("k_fold", None)
         k_fold = 1 if self.params.get("use_child_oof", False) else k_fold
         if k_fold is not None and k_fold > 0:

autogluon/core/version.py

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

{autogluon.core-1.2.1b20250414.dist-info → autogluon.core-1.2.1b20250416.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-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
@@ -38,19 +38,19 @@ License-File: ../LICENSE
 License-File: ../NOTICE
 Requires-Dist: numpy<2.3.0,>=1.25.0
 Requires-Dist: scipy<1.16,>=1.5.4
-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: pandas<2.3.0,>=2.0.0
 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.1b20250414
+Requires-Dist: autogluon.common==1.2.1b20250416
 Provides-Extra: all
-Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "all"
-Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "all"
 Requires-Dist: pyarrow>=15.0.0; extra == "all"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
+Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "all"
+Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "all"
 Provides-Extra: ray
 Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "ray"
 Provides-Extra: raytune
@@ -58,11 +58,11 @@ Requires-Dist: pyarrow>=15.0.0; extra == "raytune"
 Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "raytune"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "raytune"
 Provides-Extra: tests
-Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: types-requests; extra == "tests"
-Requires-Dist: flake8; extra == "tests"
-Requires-Dist: pytest; extra == "tests"
 Requires-Dist: pytest-mypy; extra == "tests"
+Requires-Dist: pytest; extra == "tests"
+Requires-Dist: flake8; extra == "tests"
+Requires-Dist: types-setuptools; extra == "tests"
 
 
 

{autogluon.core-1.2.1b20250414.dist-info → autogluon.core-1.2.1b20250416.dist-info}/RECORD

@@ -1,9 +1,9 @@
-autogluon.core-1.2.1b20250414-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.core-1.2.1b20250416-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
-autogluon/core/_setup_utils.py,sha256=G4IL_kbcDxhX9SOCpiy9Xk07qJIgj_HcsWBAP6w-5kM,7120
+autogluon/core/_setup_utils.py,sha256=TnDboM66MzSp0CLACo_vcNNwhORQWE8JL0kt1IYa7dI,7014
 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=mZy8mr13P4Mk82yBos9i1Hq0vzFrn8df1GDIapFERNg,91
+autogluon/core/version.py,sha256=qRiT5RKmHqutDnkFPpmNCIubktJud5v6ljMK0Td-yh0,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
@@ -40,7 +40,7 @@ autogluon/core/models/__init__.py,sha256=raRAG4wml-xvCKp3zxthda7of9NEWcpafgpvZWJ
 autogluon/core/models/_utils.py,sha256=qswE9n1ge1AJSExgstEbrZiMFmMRa4Mf5Sz8D9-XU6c,2091
 autogluon/core/models/abstract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/abstract/_tags.py,sha256=Qr_3an0ZMig24S3OwISa-nTFfWHQe3pwPTiXq4zlEec,3409
-autogluon/core/models/abstract/abstract_model.py,sha256=MgR0uB3hFQFM-NaHRBhg9e1gWNlu-zHEm2_wURWtHLs,124309
+autogluon/core/models/abstract/abstract_model.py,sha256=S4S_cYO_CZtk2tnk6-IwpXP0V7RfWYHsR9R7DmGMnPU,124310
 autogluon/core/models/abstract/abstract_nn_model.py,sha256=IId0ivO8uVvmpnK9OiM2CtPVrP1ewOaQQKtQUDtK7_k,4818
 autogluon/core/models/abstract/model_trial.py,sha256=PKEo1jfLSBCOLM42QE5VBD1u41MaVMRk31zhNhLiqTw,5035
 autogluon/core/models/dummy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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.1b20250414.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.core-1.2.1b20250414.dist-info/METADATA,sha256=snoqJCIiDBRLhF9-5yuQYO5sdVpbLza_XEo-uU5SS30,12399
-autogluon.core-1.2.1b20250414.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.core-1.2.1b20250414.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.core-1.2.1b20250414.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250414.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250414.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.core-1.2.1b20250414.dist-info/RECORD,,
+autogluon.core-1.2.1b20250416.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.core-1.2.1b20250416.dist-info/METADATA,sha256=NNkr5lZiYKwjw7GhgSsj9mdk4dg2mppSqvdDFiDHPxg,12399
+autogluon.core-1.2.1b20250416.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.core-1.2.1b20250416.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.core-1.2.1b20250416.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250416.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250416.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.core-1.2.1b20250416.dist-info/RECORD,,