autogluon.core 1.2.1b20250410__py3-none-any.whl → 1.2.1b20250411__py3-none-any.whl

autogluon/core/_setup_utils.py

@@ -17,13 +17,13 @@ PYTHON_REQUIRES = ">=3.9, <3.13"
 # Only put packages here that would otherwise appear multiple times across different module's setup.py files.
 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.1.4",  # "<{N+3}" upper cap, where N is the latest released minor version, assuming no warnings using N
+    "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.5.3",  # capping to latest version
+    "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
     "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.0.0",  # Major version cap
-    "s3fs": ">=2023.1,<2025",  # Yearly cap
+    "psutil": ">=5.7.3,<7.1.0",  # Major version cap
+    "s3fs": ">=2024.2,<2026",  # Yearly cap
     "networkx": ">=3.0,<4",  # Major version cap
     "tqdm": ">=4.38,<5",  # Major version cap
     "Pillow": ">=10.0.1,<12",  # Major version cap

autogluon/core/metrics/__init__.py

@@ -603,7 +603,7 @@ pearsonr = make_scorer("pearsonr", local_pearsonr, optimum=1.0, greater_is_bette
 
 def rmse_func(y_true, y_pred, **kwargs):
     if kwargs:
-        return sklearn.metrics.mean_squared_error(y_true, y_pred, squared=False, **kwargs)
+        return sklearn.metrics.root_mean_squared_error(y_true, y_pred, **kwargs)
     else:
         return np.sqrt(((y_true - y_pred) ** 2).mean())
 

autogluon/core/version.py

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

{autogluon.core-1.2.1b20250410.dist-info → autogluon.core-1.2.1b20250411.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250410
+Version: 1.2.1b20250411
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -36,32 +36,32 @@ Requires-Python: >=3.9, <3.13
 Description-Content-Type: text/markdown
 License-File: ../LICENSE
 License-File: ../NOTICE
-Requires-Dist: numpy<2.1.4,>=1.25.0
+Requires-Dist: numpy<2.3.0,>=1.25.0
 Requires-Dist: scipy<1.16,>=1.5.4
-Requires-Dist: scikit-learn<1.5.3,>=1.4.0
+Requires-Dist: scikit-learn<1.6.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.1b20250410
+Requires-Dist: autogluon.common==1.2.1b20250411
 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: pyarrow>=15.0.0; extra == "all"
-Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "all"
+Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "all"
+Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
+Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "all"
 Provides-Extra: ray
-Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "ray"
+Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "ray"
 Provides-Extra: raytune
 Requires-Dist: pyarrow>=15.0.0; extra == "raytune"
-Requires-Dist: ray[default,tune]<2.43,>=2.10.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: pytest; extra == "tests"
+Requires-Dist: types-requests; extra == "tests"
 Requires-Dist: pytest-mypy; extra == "tests"
 Requires-Dist: flake8; extra == "tests"
-Requires-Dist: types-requests; extra == "tests"
 Requires-Dist: types-setuptools; extra == "tests"
 
 

{autogluon.core-1.2.1b20250410.dist-info → autogluon.core-1.2.1b20250411.dist-info}/RECORD

@@ -1,9 +1,9 @@
-autogluon.core-1.2.1b20250410-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.core-1.2.1b20250411-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
-autogluon/core/_setup_utils.py,sha256=aSBdEF03KDWn4ct5m9aHRjrMxWi_qWVh-xemDtw_pQY,7022
+autogluon/core/_setup_utils.py,sha256=G4IL_kbcDxhX9SOCpiy9Xk07qJIgj_HcsWBAP6w-5kM,7120
 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=tJeST0EZCa1sWbuBSG5OURtJwcs6LfjpwjLa8Ns8AI4,91
+autogluon/core/version.py,sha256=Me6Wx0KIEaoCqLzX7xcFk0juNCDMK_1EhFkh_68We6Q,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
@@ -31,7 +31,7 @@ autogluon/core/hpo/space_converter.py,sha256=pAQpSUonRGcR8l2uyCVJCxY-bd031vlU5rU
 autogluon/core/learner/__init__.py,sha256=pAEQBcY-xq-HG2QbHdt9V4iq1RSlz6gfcxNa40abGVY,77
 autogluon/core/learner/abstract_learner.py,sha256=A4nAR-U27wInLtghQjpLdEhBgh9M_2DFz7NhoVhjuL4,5784
 autogluon/core/learning_curves/plot_curves.py,sha256=GxWeX6gvp9XlA2QwtBi9cdaxjtbY3mSngWXl-EjM8bE,3141
-autogluon/core/metrics/__init__.py,sha256=zbfjLVA7qcYwyTWAo54XZR7MklYgyOjaDZNCcMPyHPg,29837
+autogluon/core/metrics/__init__.py,sha256=7UTGdSwuRG4Wjfenyvj9RBYMtTSGAyPpy0Q_ktF-AF8,29827
 autogluon/core/metrics/classification_metrics.py,sha256=kKbq1cBMmc_gEm3AfXBCTk5KJDcrzcg-6gmQ_wONhFU,18342
 autogluon/core/metrics/quantile_metrics.py,sha256=yjnKSMuLdee4vemXIoxitCJWf_3lvE-RXlJ4V1371w4,1393
 autogluon/core/metrics/score_func.py,sha256=RIFOZMhPKYRFXBufoe5XIY1BRDr8aqGA-2zVwuTKi0A,4684
@@ -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.1b20250410.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.core-1.2.1b20250410.dist-info/METADATA,sha256=ypewiInlg9JClew9T3s7Trybj6nZkUT2C5RTAn_2j1k,12399
-autogluon.core-1.2.1b20250410.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.core-1.2.1b20250410.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.core-1.2.1b20250410.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250410.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250410.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.core-1.2.1b20250410.dist-info/RECORD,,
+autogluon.core-1.2.1b20250411.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.core-1.2.1b20250411.dist-info/METADATA,sha256=HNhXPI-rV4odhDEkTAndKMtJzI0H5TaR1QnAoOC_qrA,12399
+autogluon.core-1.2.1b20250411.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.core-1.2.1b20250411.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.core-1.2.1b20250411.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250411.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250411.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.core-1.2.1b20250411.dist-info/RECORD,,