autogluon.core 1.2.1b20250218__tar.gz → 1.2.1b20250220__tar.gz

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250218
+Version: 1.2.1b20250220
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/setup.py

@@ -48,11 +48,11 @@ install_requires = (
 
 extras_require = {
     "ray": [
-        "ray[default]>=2.10.0,<2.41",  # sync with common/src/autogluon/common/utils/try_import.py
+        "ray[default]>=2.10.0,<2.43",  # 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.41",  # sync with common/src/autogluon/common/utils/try_import.py
+        "ray[default,tune]>=2.10.0,<2.43",  # 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.1b20250218 → autogluon.core-1.2.1b20250220}/src/autogluon/core/models/abstract/abstract_model.py

@@ -1632,9 +1632,7 @@ class AbstractModel:
         path = self.path_root
         problem_type = self.problem_type
         eval_metric = self.eval_metric
-        hyperparameters = self._user_params.copy()
-        if self._user_params_aux:
-            hyperparameters[AG_ARGS_FIT] = self._user_params_aux.copy()
+        hyperparameters = self.get_hyperparameters_init()
 
         args = dict(
             path=path,
@@ -1646,6 +1644,20 @@ class AbstractModel:
 
         return args
 
+    def get_hyperparameters_init(self) -> dict:
+        """
+
+        Returns
+        -------
+        hyperparameters: dict
+            The dictionary of user specified hyperparameters for the model.
+
+        """
+        hyperparameters = self._user_params.copy()
+        if self._user_params_aux:
+            hyperparameters[AG_ARGS_FIT] = self._user_params_aux.copy()
+        return hyperparameters
+
     def convert_to_template(self):
         """
         After calling this function, returned model should be able to be fit as if it was new, as well as deep-copied.
@@ -2227,7 +2239,7 @@ class AbstractModel:
         # TODO: Report errors?
         shutil.rmtree(path=model_path, ignore_errors=True)
 
-    def get_info(self) -> dict:
+    def get_info(self, include_feature_metadata: bool = True) -> dict:
         """
         Returns a dictionary of numerous fields describing the model.
         """
@@ -2243,6 +2255,7 @@ class AbstractModel:
             "predict_time": self.predict_time,
             "val_score": self.val_score,
             "hyperparameters": self.params,
+            "hyperparameters_user": self.get_hyperparameters_init(),
             "hyperparameters_fit": self.params_trained,  # TODO: Explain in docs that this is for hyperparameters that differ in final model from original hyperparameters, such as epochs (from early stopping)
             "hyperparameters_nondefault": self.nondefault_params,
             AG_ARGS_FIT: self.get_params_aux_info(),
@@ -2260,6 +2273,8 @@ class AbstractModel:
         }
         if self._is_fit_metadata_registered:
             info.update(self._fit_metadata)
+        if not include_feature_metadata:
+            info.pop("feature_metadata")
         return info
 
     def get_params_aux_info(self) -> dict:

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/src/autogluon/core/models/ensemble/bagged_ensemble_model.py

@@ -1192,6 +1192,28 @@ class BaggedEnsembleModel(AbstractModel):
         init_args.pop("problem_type")
         return init_args
 
+    def get_hyperparameters_init_child(self, include_ag_args_ensemble: bool = False, child_model: AbstractModel = None) -> dict:
+        """
+
+        Returns
+        -------
+        hyperparameters: dict
+            The dictionary of user specified hyperparameters for the model.
+
+        """
+        if child_model is None:
+            if self.n_children > 0:
+                child_model = self.load_child(self.models[0])
+            else:
+                child_model = self._get_model_base()
+        hyperparameters_child = child_model.get_hyperparameters_init()
+        if include_ag_args_ensemble:
+            hyperparameters_self = self.get_hyperparameters_init()
+            if hyperparameters_self:
+                hyperparameters_child["ag_args_ensemble"] = hyperparameters_self
+
+        return hyperparameters_child
+
     def convert_to_template_child(self):
         return self._get_model_base().convert_to_template()
 
@@ -1409,9 +1431,9 @@ class BaggedEnsembleModel(AbstractModel):
                 model_names.append(model.name)
         return model_names
 
-    def get_info(self):
-        info = super().get_info()
-        children_info = self._get_child_info()
+    def get_info(self, include_feature_metadata: bool = True):
+        info = super().get_info(include_feature_metadata=include_feature_metadata)
+        children_info = self._get_child_info(include_feature_metadata=include_feature_metadata)
         child_memory_sizes = [child["memory_size"] for child in children_info.values()]
         sum_memory_size_child = sum(child_memory_sizes)
         if child_memory_sizes:
@@ -1432,6 +1454,7 @@ class BaggedEnsembleModel(AbstractModel):
             child_model = self._get_model_base()
         child_hyperparameters = child_model.params
         child_ag_args_fit = child_model.params_aux
+        child_hyperparameters_user = self.get_hyperparameters_init_child(include_ag_args_ensemble=False, child_model=child_model)
 
         bagged_info = dict(
             child_model_type=self._child_type.__name__,
@@ -1448,6 +1471,7 @@ class BaggedEnsembleModel(AbstractModel):
             max_memory_size=max_memory_size,  # Memory used when all children are loaded into memory at once.
             min_memory_size=min_memory_size,  # Memory used when only the largest child is loaded into memory.
             child_hyperparameters=child_hyperparameters,
+            child_hyperparameters_user=child_hyperparameters_user,
             child_hyperparameters_fit=self._get_compressed_params_trained(),
             child_ag_args_fit=child_ag_args_fit,
         )
@@ -1480,14 +1504,16 @@ class BaggedEnsembleModel(AbstractModel):
         # memory is checked downstream on the child model
         pass
 
-    def _get_child_info(self):
+    def _get_child_info(self, include_feature_metadata: bool = True):
         child_info_dict = dict()
         for model in self.models:
             if isinstance(model, str):
                 child_path = self.create_contexts(os.path.join(self.path, model))
                 child_info_dict[model] = self._child_type.load_info(child_path)
+                if not include_feature_metadata:
+                    child_info_dict[model].pop("feature_metadata", None)
             else:
-                child_info_dict[model.name] = model.get_info()
+                child_info_dict[model.name] = model.get_info(include_feature_metadata=include_feature_metadata)
         return child_info_dict
 
     def _construct_empty_oof(self, X, y):

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/src/autogluon/core/models/ensemble/stacker_ensemble_model.py

@@ -309,8 +309,8 @@ class StackerEnsembleModel(BaggedEnsembleModel):
             model = model_type.load(model_path)
         return model
 
-    def get_info(self):
-        info = super().get_info()
+    def get_info(self, **kwargs):
+        info = super().get_info(**kwargs)
         stacker_info = dict(
             num_base_models=len(self.base_model_names),
             base_model_names=self.base_model_names,

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/src/autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py

@@ -111,8 +111,8 @@ class GreedyWeightedEnsembleModel(AbstractModel):
         model_weight_dict = {self.base_model_names[i]: self.weights_[i] for i in range(num_models)}
         return model_weight_dict
 
-    def get_info(self):
-        info = super().get_info()
+    def get_info(self, **kwargs):
+        info = super().get_info(**kwargs)
         info["model_weights"] = self._get_model_weights()
         return info
 

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/src/autogluon/core/version.py

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

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/src/autogluon.core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250218
+Version: 1.2.1b20250220
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.core-1.2.1b20250218 → autogluon.core-1.2.1b20250220}/src/autogluon.core.egg-info/requires.txt

@@ -7,25 +7,25 @@ tqdm<5,>=4.38
 requests
 matplotlib<3.11,>=3.7.0
 boto3<2,>=1.10
-autogluon.common==1.2.1b20250218
+autogluon.common==1.2.1b20250220
 
 [all]
+ray[default,tune]<2.43,>=2.10.0
+ray[default]<2.43,>=2.10.0
 pyarrow>=15.0.0
-ray[default]<2.41,>=2.10.0
 hyperopt<0.2.8,>=0.2.7
-ray[default,tune]<2.41,>=2.10.0
 
 [ray]
-ray[default]<2.41,>=2.10.0
+ray[default]<2.43,>=2.10.0
 
 [raytune]
 pyarrow>=15.0.0
-ray[default,tune]<2.41,>=2.10.0
+ray[default,tune]<2.43,>=2.10.0
 hyperopt<0.2.8,>=0.2.7
 
 [tests]
+pytest-mypy
+types-setuptools
 flake8
 types-requests
-types-setuptools
 pytest
-pytest-mypy