autogluon.core 1.2.1b20250414__tar.gz → 1.2.1b20250416__tar.gz

{autogluon.core-1.2.1b20250414 → autogluon.core-1.2.1b20250416}/PKG-INFO

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

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

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

{autogluon.core-1.2.1b20250414 → autogluon.core-1.2.1b20250416}/src/autogluon.core.egg-info/requires.txt

@@ -1,19 +1,19 @@
 numpy<2.3.0,>=1.25.0
 scipy<1.16,>=1.5.4
-scikit-learn<1.6.0,>=1.4.0
+scikit-learn<1.7.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.1b20250414
+autogluon.common==1.2.1b20250416
 
 [all]
-ray[default]<2.45,>=2.10.0
-ray[default,tune]<2.45,>=2.10.0
 pyarrow>=15.0.0
 hyperopt<0.2.8,>=0.2.7
+ray[default,tune]<2.45,>=2.10.0
+ray[default]<2.45,>=2.10.0
 
 [ray]
 ray[default]<2.45,>=2.10.0
@@ -24,8 +24,8 @@ ray[default,tune]<2.45,>=2.10.0
 hyperopt<0.2.8,>=0.2.7
 
 [tests]
-types-setuptools
 types-requests
-flake8
-pytest
 pytest-mypy
+pytest
+flake8
+types-setuptools