autogluon.core 1.2.1b20250410__tar.gz → 1.2.1b20250411__tar.gz

{autogluon.core-1.2.1b20250410 → autogluon.core-1.2.1b20250411}/PKG-INFO

@@ -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

{autogluon.core-1.2.1b20250410 → autogluon.core-1.2.1b20250411}/setup.py

@@ -48,11 +48,11 @@ install_requires = (
 
 extras_require = {
     "ray": [
-        "ray[default]>=2.10.0,<2.43",  # sync with common/src/autogluon/common/utils/try_import.py
+        "ray[default]>=2.10.0,<2.45",  # sync with common/src/autogluon/common/utils/try_import.py
     ],
     "raytune": [
         "pyarrow>=15.0.0",  # cap Pyarrow to fix source installation - https://github.com/autogluon/autogluon/issues/4519
-        "ray[default,tune]>=2.10.0,<2.43",  # sync with common/src/autogluon/common/utils/try_import.py
+        "ray[default,tune]>=2.10.0,<2.45",  # sync with common/src/autogluon/common/utils/try_import.py
         # TODO: consider alternatives as hyperopt is not actively maintained.
         "hyperopt>=0.2.7,<0.2.8",  # This is needed for the bayes search to work.
         # 'GPy>=1.10.0,<1.11.0'  # TODO: Enable this once PBT/PB2 are supported by ray lightning

{autogluon.core-1.2.1b20250410 → autogluon.core-1.2.1b20250411}/src/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-1.2.1b20250410 → autogluon.core-1.2.1b20250411}/src/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-1.2.1b20250410 → autogluon.core-1.2.1b20250411}/src/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 → autogluon.core-1.2.1b20250411}/src/autogluon.core.egg-info/PKG-INFO

@@ -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

{autogluon.core-1.2.1b20250410 → autogluon.core-1.2.1b20250411}/src/autogluon.core.egg-info/requires.txt

@@ -1,31 +1,31 @@
-numpy<2.1.4,>=1.25.0
+numpy<2.3.0,>=1.25.0
 scipy<1.16,>=1.5.4
-scikit-learn<1.5.3,>=1.4.0
+scikit-learn<1.6.0,>=1.4.0
 networkx<4,>=3.0
 pandas<2.3.0,>=2.0.0
 tqdm<5,>=4.38
 requests
 matplotlib<3.11,>=3.7.0
 boto3<2,>=1.10
-autogluon.common==1.2.1b20250410
+autogluon.common==1.2.1b20250411
 
 [all]
-hyperopt<0.2.8,>=0.2.7
-ray[default,tune]<2.43,>=2.10.0
 pyarrow>=15.0.0
-ray[default]<2.43,>=2.10.0
+ray[default,tune]<2.45,>=2.10.0
+hyperopt<0.2.8,>=0.2.7
+ray[default]<2.45,>=2.10.0
 
 [ray]
-ray[default]<2.43,>=2.10.0
+ray[default]<2.45,>=2.10.0
 
 [raytune]
 pyarrow>=15.0.0
-ray[default,tune]<2.43,>=2.10.0
+ray[default,tune]<2.45,>=2.10.0
 hyperopt<0.2.8,>=0.2.7
 
 [tests]
 pytest
+types-requests
 pytest-mypy
 flake8
-types-requests
 types-setuptools