autogluon.tabular 1.2.1b20250205__py3-none-any.whl → 1.2.1b20250207__py3-none-any.whl

autogluon/tabular/predictor/predictor.py

@@ -44,7 +44,6 @@ from autogluon.core.problem_type import problem_type_info
 from autogluon.core.pseudolabeling.pseudolabeling import filter_ensemble_pseudo, filter_pseudo
 from autogluon.core.scheduler.scheduler_factory import scheduler_factory
 from autogluon.core.stacked_overfitting.utils import check_stacked_overfitting_from_leaderboard
-from autogluon.core.trainer import AbstractTrainer
 from autogluon.core.utils import get_pred_from_proba_df, plot_performance_vs_trials, plot_summary_of_models, plot_tabular_models
 from autogluon.core.utils.decorators import apply_presets
 from autogluon.core.utils.loaders import load_pkl, load_str
@@ -55,6 +54,7 @@ from ..configs.feature_generator_presets import get_default_feature_generator
 from ..configs.hyperparameter_configs import get_hyperparameter_config
 from ..configs.presets_configs import tabular_presets_alias, tabular_presets_dict
 from ..learner import AbstractTabularLearner, DefaultLearner
+from ..trainer.abstract_trainer import AbstractTabularTrainer
 from ..trainer.model_presets.presets import MODEL_TYPES
 from ..version import __version__
 from ._deprecated_methods import TabularPredictorDeprecatedMixin
@@ -163,8 +163,8 @@ class TabularPredictor(TabularPredictorDeprecatedMixin):
             cache_data : bool, default = True
                 When enabled, the training and validation data are saved to disk for future reuse.
                 Enables advanced functionality in predictor such as `fit_extra()` and feature importance calculation on the original data.
-            trainer_type : AbstractTrainer, default = AutoTrainer
-                A class inheriting from `AbstractTrainer` that controls training/ensembling of many models.
+            trainer_type : AbstractTabularTrainer, default = AutoTrainer
+                A class inheriting from `AbstractTabularTrainer` that controls training/ensembling of many models.
                 If you don't know what this is, keep it as the default.
     """
 
@@ -223,7 +223,7 @@ class TabularPredictor(TabularPredictorDeprecatedMixin):
             **learner_kwargs,
         )
         self._learner_type = type(self._learner)
-        self._trainer: AbstractTrainer = None
+        self._trainer: AbstractTabularTrainer = None
         self._sub_fits: list[str] = []
         self._stacked_overfitting_occurred: bool | None = None
         self._fit_strategy = None
@@ -828,7 +828,7 @@ class TabularPredictor(TabularPredictorDeprecatedMixin):
                 again with the pruned set of features, and updates input feature lists for models whose validation score improved.
                 If None, do not perform feature pruning. If empty dictionary, perform feature pruning with default configurations.
                 For valid dictionary keys, refer to :class:`autogluon.core.utils.feature_selection.FeatureSelector` and
-                `autogluon.core.trainer.abstract_trainer.AbstractTrainer._proxy_model_feature_prune` documentation.
+                `autogluon.core.trainer.abstract_trainer.AbstractTabularTrainer._proxy_model_feature_prune` documentation.
                 To force all models to work with the pruned set of features, set force_prune=True in the dictionary.
             ag_args : dict, default = None
                 Keyword arguments to pass to all models (i.e. common hyperparameters shared by all AutoGluon models).
@@ -4665,7 +4665,7 @@ class TabularPredictor(TabularPredictorDeprecatedMixin):
         self._learner_type = type(self._learner)
         if self._learner.trainer_path is not None:
             self._learner.persist_trainer(low_memory=True)
-            self._trainer: AbstractTrainer = self._learner.load_trainer()  # Trainer object
+            self._trainer: AbstractTabularTrainer = self._learner.load_trainer()  # Trainer object
 
     @classmethod
     def _load_version_file(cls, path: str) -> str: