autogluon.tabular 1.3.2b20250715__py3-none-any.whl → 1.3.2b20250716__py3-none-any.whl

autogluon/tabular/models/mitra/sklearn_interface.py

@@ -1,20 +1,20 @@
-import numpy as np
 import time
-import torch
-import pandas as pd
-
 from pathlib import Path
+
+import numpy as np
+import pandas as pd
+import torch
 from sklearn.base import BaseEstimator, ClassifierMixin, RegressorMixin
 
-from ._internal.data.dataset_split import make_stratified_dataset_split
 from ._internal.config.config_run import ConfigRun
+from ._internal.config.enums import ModelName
 from ._internal.core.trainer_finetune import TrainerFinetune
+from ._internal.data.dataset_split import make_stratified_dataset_split
 from ._internal.models.tab2d import Tab2D
-from ._internal.config.enums import ModelName
 
 # Hyperparameter search space
 DEFAULT_FINE_TUNE = True # [True, False]
-DEFAULT_FINE_TUNE_STEPS = 50 # [50, 60, 70, 80, 90, 100] 
+DEFAULT_FINE_TUNE_STEPS = 50 # [50, 60, 70, 80, 90, 100]
 DEFAULT_CLS_METRIC = 'log_loss' # ['log_loss', 'accuracy', 'auc']
 DEFAULT_REG_METRIC = 'mse' # ['mse', 'mae', 'rmse', 'r2']
 SHUFFLE_CLASSES = False # [True, False]
@@ -32,7 +32,17 @@ DEFAULT_REG_MODEL = 'autogluon/mitra-regressor'
 # Constants
 SEED = 0
 DEFAULT_MODEL_TYPE = "Tab2D"
-DEFAULT_DEVICE = "cuda"
+
+def _get_default_device():
+    """Get the best available device for the current system."""
+    if torch.cuda.is_available():
+        return "cuda"
+    elif hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
+        return "mps"  # Apple silicon
+    else:
+        return "cpu"
+
+DEFAULT_DEVICE = _get_default_device()
 DEFAULT_ENSEMBLE = 1
 DEFAULT_DIM = 512
 DEFAULT_LAYERS = 12
@@ -43,13 +53,13 @@ USE_HF = True  # Use Hugging Face pretrained models if available
 
 class MitraBase(BaseEstimator):
     """Base class for Mitra models with common functionality."""
-    
-    def __init__(self, 
-            model_type=DEFAULT_MODEL_TYPE, 
-            n_estimators=DEFAULT_ENSEMBLE, 
-            device=DEFAULT_DEVICE, 
+
+    def __init__(self,
+            model_type=DEFAULT_MODEL_TYPE,
+            n_estimators=DEFAULT_ENSEMBLE,
+            device=DEFAULT_DEVICE,
             fine_tune=DEFAULT_FINE_TUNE,
-            fine_tune_steps=DEFAULT_FINE_TUNE_STEPS, 
+            fine_tune_steps=DEFAULT_FINE_TUNE_STEPS,
             metric=DEFAULT_CLS_METRIC,
             state_dict=None,
             hf_general_model=DEFAULT_GENERAL_MODEL,
@@ -155,7 +165,7 @@ class MitraBase(BaseEstimator):
 
         return cfg, Tab2D
 
-    
+
     def _split_data(self, X, y):
         """Split data into training and validation sets."""
         if hasattr(self, 'task') and self.task == 'classification':
@@ -165,7 +175,7 @@ class MitraBase(BaseEstimator):
             val_indices = np.random.choice(range(len(X)), int(DEFAULT_VALIDATION_SPLIT * len(X)), replace=False).tolist()
             train_indices = [i for i in range(len(X)) if i not in val_indices]
             return X[train_indices], X[val_indices], y[train_indices], y[val_indices]
-    
+
     def _train_ensemble(self, X_train, y_train, X_valid, y_valid, task, dim_output, n_classes=0, time_limit=None):
         """Train the ensemble of models."""
 
@@ -175,7 +185,7 @@ class MitraBase(BaseEstimator):
         while not (success and cfg.hyperparams["max_samples_support"] > 0 and cfg.hyperparams["max_samples_query"] > 0):
             try:
                 self.trainers.clear()
-                
+
                 self.train_time = 0
                 for _ in range(self.n_estimators):
                     if USE_HF:
@@ -212,7 +222,7 @@ class MitraBase(BaseEstimator):
 
                     self.trainers.append(trainer)
                     self.train_time += end_time - start_time
-                    
+
                     success = True
 
             except torch.cuda.OutOfMemoryError:
@@ -233,24 +243,24 @@ class MitraBase(BaseEstimator):
                     )
                     print(f"Reducing max_samples_query from {cfg.hyperparams['max_samples_query'] * 2}"
                           f"to {cfg.hyperparams['max_samples_query']} due to OOM error.")
-                    
+
         if not success:
             raise RuntimeError(
-                f"Failed to train Mitra model after multiple attempts due to out of memory error."
+                "Failed to train Mitra model after multiple attempts due to out of memory error."
             )
-        
+
         return self
 
 
 class MitraClassifier(MitraBase, ClassifierMixin):
     """Classifier implementation of Mitra model."""
 
-    def __init__(self, 
-            model_type=DEFAULT_MODEL_TYPE, 
-            n_estimators=DEFAULT_ENSEMBLE, 
-            device=DEFAULT_DEVICE, 
+    def __init__(self,
+            model_type=DEFAULT_MODEL_TYPE,
+            n_estimators=DEFAULT_ENSEMBLE,
+            device=DEFAULT_DEVICE,
             fine_tune=DEFAULT_FINE_TUNE,
-            fine_tune_steps=DEFAULT_FINE_TUNE_STEPS, 
+            fine_tune_steps=DEFAULT_FINE_TUNE_STEPS,
             metric=DEFAULT_CLS_METRIC,
             state_dict=None,
             patience=PATIENCE,
@@ -265,12 +275,12 @@ class MitraClassifier(MitraBase, ClassifierMixin):
         ):
         """Initialize the classifier."""
         super().__init__(
-            model_type, 
-            n_estimators, 
-            device, 
+            model_type,
+            n_estimators,
+            device,
             fine_tune,
             fine_tune_steps,
-            metric, 
+            metric,
             state_dict,
             patience=patience,
             lr=lr,
@@ -283,7 +293,7 @@ class MitraClassifier(MitraBase, ClassifierMixin):
             seed=seed,
         )
         self.task = 'classification'
-    
+
     def fit(self, X, y, X_val = None, y_val = None, time_limit = None):
         """
         Fit the ensemble of models.
@@ -367,12 +377,12 @@ class MitraClassifier(MitraBase, ClassifierMixin):
 class MitraRegressor(MitraBase, RegressorMixin):
     """Regressor implementation of Mitra model."""
 
-    def __init__(self, 
-            model_type=DEFAULT_MODEL_TYPE, 
-            n_estimators=DEFAULT_ENSEMBLE, 
-            device=DEFAULT_DEVICE, 
+    def __init__(self,
+            model_type=DEFAULT_MODEL_TYPE,
+            n_estimators=DEFAULT_ENSEMBLE,
+            device=DEFAULT_DEVICE,
             fine_tune=DEFAULT_FINE_TUNE,
-            fine_tune_steps=DEFAULT_FINE_TUNE_STEPS, 
+            fine_tune_steps=DEFAULT_FINE_TUNE_STEPS,
             metric=DEFAULT_REG_METRIC,
             state_dict=None,
             patience=PATIENCE,
@@ -387,12 +397,12 @@ class MitraRegressor(MitraBase, RegressorMixin):
         ):
         """Initialize the regressor."""
         super().__init__(
-            model_type, 
-            n_estimators, 
-            device, 
+            model_type,
+            n_estimators,
+            device,
             fine_tune,
             fine_tune_steps,
-            metric, 
+            metric,
             state_dict,
             patience=patience,
             lr=lr,
@@ -457,6 +467,6 @@ class MitraRegressor(MitraBase, RegressorMixin):
         """
         if isinstance(X, pd.DataFrame):
             X = X.values
-        
+
         preds = [trainer.predict(self.X, self.y, X) for trainer in self.trainers]
-        return sum(preds) / len(preds)  # Averaging ensemble predictions
+        return sum(preds) / len(preds)  # Averaging ensemble predictions

autogluon/tabular/models/realmlp/realmlp_model.py

@@ -55,7 +55,12 @@ class RealMLPModel(AbstractModel):
         self._bool_to_cat = None
 
     def get_model_cls(self, default_hyperparameters: Literal["td", "td_s"] = "td"):
-        from pytabkit import RealMLP_TD_Classifier, RealMLP_TD_Regressor, RealMLP_TD_S_Classifier, RealMLP_TD_S_Regressor
+        from pytabkit import (
+            RealMLP_TD_Classifier,
+            RealMLP_TD_Regressor,
+            RealMLP_TD_S_Classifier,
+            RealMLP_TD_S_Regressor,
+        )
 
         assert default_hyperparameters in ["td", "td_s"]
         if self.problem_type in ['binary', 'multiclass']:
@@ -267,9 +272,11 @@ class RealMLPModel(AbstractModel):
         return self.eval_metric
 
     def _get_default_resources(self) -> tuple[int, int]:
-        # only_physical_cores=True is faster in training
+        # Use only physical cores for better performance based on benchmarks
         num_cpus = ResourceManager.get_cpu_count(only_physical_cores=True)
-        num_gpus = min(ResourceManager.get_gpu_count_torch(), 1)
+        # Only request GPU if CUDA is available (RealMLP doesn't support MPS)
+        import torch
+        num_gpus = 1 if torch.cuda.is_available() else 0
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:
@@ -345,3 +352,4 @@ class RealMLPModel(AbstractModel):
         #  How to mirror RealMLP learning rate scheduler while forcing stopping at a specific epoch?
         tags = {"can_refit_full": False}
         return tags
+        return tags

autogluon/tabular/models/tabicl/tabicl_model.py

@@ -109,8 +109,11 @@ class TabICLModel(AbstractModel):
         return ["binary", "multiclass"]
 
     def _get_default_resources(self) -> tuple[int, int]:
+        # Use only physical cores for better performance based on benchmarks
         num_cpus = ResourceManager.get_cpu_count(only_physical_cores=True)
-        num_gpus = min(ResourceManager.get_gpu_count_torch(), 1)
+        # Only request GPU if CUDA is available (TabICL doesn't support MPS)
+        import torch
+        num_gpus = 1 if torch.cuda.is_available() else 0
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:

autogluon/tabular/models/tabm/_tabm_internal.py

@@ -12,13 +12,14 @@ import numpy as np
 import pandas as pd
 import scipy
 import torch
-from autogluon.core.metrics import compute_metric
 from sklearn.base import BaseEstimator, TransformerMixin
 from sklearn.impute import SimpleImputer
 from sklearn.pipeline import Pipeline
 from sklearn.preprocessing import OrdinalEncoder, QuantileTransformer
 from sklearn.utils.validation import check_is_fitted
 
+from autogluon.core.metrics import compute_metric
+
 from . import rtdl_num_embeddings, tabm_reference
 from .tabm_reference import make_parameter_groups
 
@@ -468,7 +469,7 @@ class TabMImplementation:
                 best = {"val": val_score, "epoch": epoch}
                 remaining_patience = patience
                 with torch.no_grad():
-                    for bp, p in zip(best_params, model.parameters(), strict=False):
+                    for bp, p in zip(best_params, model.parameters()):
                         bp.copy_(p)
             else:
                 remaining_patience -= 1
@@ -481,7 +482,7 @@ class TabMImplementation:
 
         logger.log(15, "Restoring best model")
         with torch.no_grad():
-            for bp, p in zip(best_params, model.parameters(), strict=False):
+            for bp, p in zip(best_params, model.parameters()):
                 p.copy_(bp)
 
         self.model_ = model

autogluon/tabular/models/tabm/tabm_model.py

@@ -19,6 +19,7 @@ import logging
 import time
 
 import pandas as pd
+
 from autogluon.common.utils.resource_utils import ResourceManager
 from autogluon.core.models import AbstractModel
 from autogluon.tabular import __version__
@@ -55,8 +56,9 @@ class TabMModel(AbstractModel):
 
         try:
             # imports various dependencies such as torch
-            from ._tabm_internal import TabMImplementation
             from torch.cuda import is_available
+
+            from ._tabm_internal import TabMImplementation
         except ImportError as err:
             logger.log(
                 40,
@@ -146,9 +148,11 @@ class TabMModel(AbstractModel):
         return self.eval_metric
 
     def _get_default_resources(self) -> tuple[int, int]:
-        # only_physical_cores=True is faster in training
+        # Use only physical cores for better performance based on benchmarks
         num_cpus = ResourceManager.get_cpu_count(only_physical_cores=True)
-        num_gpus = min(ResourceManager.get_gpu_count_torch(), 1)
+        # Only request GPU if CUDA is available (TabM doesn't support other accelerators such as MPS)
+        import torch
+        num_gpus = 1 if torch.cuda.is_available() else 0
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:

autogluon/tabular/models/tabm/tabm_reference.py

@@ -4,7 +4,7 @@
 # The minimum required versions of the dependencies are specified in README.md.
 
 import itertools
-from typing import Any, Literal
+from typing import Any, Literal, Union
 
 import torch
 import torch.nn as nn
@@ -159,9 +159,9 @@ class LinearEfficientEnsemble(nn.Module):
     avoids the term "adapter".
     """
 
-    r: None | Tensor
-    s: None | Tensor
-    bias: None | Tensor
+    r: Union[None, Tensor]
+    s: Union[None, Tensor]
+    bias: Union[None, Tensor]
 
     def __init__(
         self,
@@ -259,8 +259,8 @@ class MLP(nn.Module):
     def __init__(
         self,
         *,
-        d_in: None | int = None,
-        d_out: None | int = None,
+        d_in: Union[None, int] = None,
+        d_out: Union[None, int] = None,
         n_blocks: int,
         d_block: int,
         dropout: float,
@@ -386,19 +386,21 @@ def default_zero_weight_decay_condition(
     del module_name, parameter
     return parameter_name.endswith('bias') or isinstance(
         module,
-        nn.BatchNorm1d
-        | nn.LayerNorm
-        | nn.InstanceNorm1d
-        | rtdl_num_embeddings.LinearEmbeddings
-        | rtdl_num_embeddings.LinearReLUEmbeddings
-        | _Periodic,
+        (
+            nn.BatchNorm1d,
+            nn.LayerNorm,
+            nn.InstanceNorm1d,
+            rtdl_num_embeddings.LinearEmbeddings,
+            rtdl_num_embeddings.LinearReLUEmbeddings,
+            _Periodic,
+        ),
     )
 
 
 def make_parameter_groups(
     module: nn.Module,
     zero_weight_decay_condition=default_zero_weight_decay_condition,
-    custom_groups: None | list[dict[str, Any]] = None,
+    custom_groups: Union[None, list[dict[str, Any]]] = None,
 ) -> list[dict[str, Any]]:
     if custom_groups is None:
         custom_groups = []
@@ -439,10 +441,10 @@ class Model(nn.Module):
         *,
         n_num_features: int,
         cat_cardinalities: list[int],
-        n_classes: None | int,
+        n_classes: Union[None, int],
         backbone: dict,
-        bins: None | list[Tensor],  # For piecewise-linear encoding/embeddings.
-        num_embeddings: None | dict = None,
+        bins: Union[None, list[Tensor]],  # For piecewise-linear encoding/embeddings.
+        num_embeddings: Union[None, dict] = None,
         arch_type: Literal[
             # Plain feed-forward network without any kind of ensembling.
             'plain',
@@ -464,7 +466,7 @@ class Model(nn.Module):
             # This variant was not used in the paper.
             'tabm-mini-normal',
         ],
-        k: None | int = None,
+        k: Union[None, int] = None,
         share_training_batches: bool = True,
     ) -> None:
         # >>> Validate arguments.
@@ -593,7 +595,7 @@ class Model(nn.Module):
         self.share_training_batches = share_training_batches
 
     def forward(
-        self, x_num: None | Tensor = None, x_cat: None | Tensor = None
+        self, x_num: Union[None, Tensor] = None, x_cat: Union[None, Tensor] = None
     ) -> Tensor:
         x = []
         if x_num is not None:
@@ -624,4 +626,4 @@ class Model(nn.Module):
             # with the rest of the script (loss, metrics, predictions, ...).
             # (B, D_OUT) -> (B, 1, D_OUT)
             x = x[:, None]
-        return x
+        return x

autogluon/tabular/models/tabpfnv2/tabpfnv2_model.py

@@ -16,11 +16,12 @@ from typing import TYPE_CHECKING, Any
 
 import numpy as np
 import scipy
+from sklearn.preprocessing import PowerTransformer
+
 from autogluon.common.utils.resource_utils import ResourceManager
 from autogluon.core.models import AbstractModel
 from autogluon.features.generators import LabelEncoderFeatureGenerator
 from autogluon.tabular import __version__
-from sklearn.preprocessing import PowerTransformer
 
 if TYPE_CHECKING:
     import pandas as pd
@@ -243,10 +244,7 @@ class TabPFNV2Model(AbstractModel):
         n_ensemble_repeats = hps.pop("n_ensemble_repeats", None)
         model_is_rf_pfn = hps.pop("model_type", "no") == "dt_pfn"
         if model_is_rf_pfn:
-            from .rfpfn import (
-                RandomForestTabPFNClassifier,
-                RandomForestTabPFNRegressor,
-            )
+            from .rfpfn import RandomForestTabPFNClassifier, RandomForestTabPFNRegressor
 
             hps["n_estimators"] = 1
             rf_model_base = (
@@ -272,18 +270,21 @@ class TabPFNV2Model(AbstractModel):
     def _log_license(self, device: str):
         global _HAS_LOGGED_TABPFN_LICENSE
         if not _HAS_LOGGED_TABPFN_LICENSE:
-            logger.log(20, f"\tBuilt with PriorLabs-TabPFN")  # Aligning with TabPFNv2 license requirements
+            logger.log(20, "\tBuilt with PriorLabs-TabPFN")  # Aligning with TabPFNv2 license requirements
             if device == "cpu":
                 logger.log(
                     20,
-                    f"\tRunning TabPFNv2 on CPU. This can be very slow. "
-                    f"It is recommended to run TabPFNv2 on a GPU."
+                    "\tRunning TabPFNv2 on CPU. This can be very slow. "
+                    "It is recommended to run TabPFNv2 on a GPU."
                 )
             _HAS_LOGGED_TABPFN_LICENSE = True  # Avoid repeated logging
 
     def _get_default_resources(self) -> tuple[int, int]:
+        # Use only physical cores for better performance based on benchmarks
         num_cpus = ResourceManager.get_cpu_count(only_physical_cores=True)
-        num_gpus = min(ResourceManager.get_gpu_count_torch(), 1)
+        # Only request GPU if CUDA is available (TabPFNV2 doesn't support other accelerators such as MPS)
+        import torch
+        num_gpus = 1 if torch.cuda.is_available() else 0
         return num_cpus, num_gpus
 
     def _set_default_params(self):

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.3.2b20250715"
+__version__ = "1.3.2b20250716"
 __lite__ = False

{autogluon.tabular-1.3.2b20250715.dist-info → autogluon.tabular-1.3.2b20250716.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.3.2b20250715
+Version: 1.3.2b20250716
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,20 +41,20 @@ Requires-Dist: scipy<1.17,>=1.5.4
 Requires-Dist: pandas<2.4.0,>=2.0.0
 Requires-Dist: scikit-learn<1.8.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.3.2b20250715
-Requires-Dist: autogluon.features==1.3.2b20250715
+Requires-Dist: autogluon.core==1.3.2b20250716
+Requires-Dist: autogluon.features==1.3.2b20250716
 Provides-Extra: all
-Requires-Dist: spacy<3.9; extra == "all"
-Requires-Dist: torch<2.8,>=2.2; extra == "all"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
-Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: autogluon.core[all]==1.3.2b20250716; extra == "all"
+Requires-Dist: pytabkit<1.6,>=1.5; extra == "all"
 Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: autogluon.core[all]==1.3.2b20250715; extra == "all"
+Requires-Dist: torch<2.8,>=2.2; extra == "all"
+Requires-Dist: spacy<3.9; extra == "all"
+Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
+Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: pytabkit<1.6,>=1.5; extra == "all"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -72,7 +72,7 @@ Requires-Dist: einx; extra == "mitra"
 Requires-Dist: omegaconf; extra == "mitra"
 Requires-Dist: transformers; extra == "mitra"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.3.2b20250715; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.3.2b20250716; extra == "ray"
 Provides-Extra: realmlp
 Requires-Dist: pytabkit<1.6,>=1.5; extra == "realmlp"
 Provides-Extra: skex

{autogluon.tabular-1.3.2b20250715.dist-info → autogluon.tabular-1.3.2b20250716.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.3.2b20250715-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.3.2b20250716-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=__YNhWg03UP-7oVi2ZS_VYHrVX7DWpIkJf61hjFPiDw,91
+autogluon/tabular/version.py,sha256=bPJmn8IYFHHx5wv_RbAGbiMuX0PyNDOc2m8l8Fn-Gx8,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/config_helper.py,sha256=JsdVGmpcYL88GPKBznPtqJ1sGaByOSvLn7KWU-HyVoQ,21085
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
@@ -68,27 +68,33 @@ autogluon/tabular/models/lr/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TIm
 autogluon/tabular/models/lr/hyperparameters/parameters.py,sha256=Hr5YC13zjbt3CfCbzGj8iXUIuDn-Q7FvDT2uSuiSVlM,1414
 autogluon/tabular/models/lr/hyperparameters/searchspaces.py,sha256=Igywc-B6qJ9EBLdasrDhW-Ot5FGirIzbXLwv5HRe5Xo,276
 autogluon/tabular/models/mitra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/mitra/mitra_model.py,sha256=3y9WiTHWPo8ORx7Q8w_g1-oKePxpGqG7vDWhd5OCKNA,7409
-autogluon/tabular/models/mitra/sklearn_interface.py,sha256=0Gqhr9GZlkOBXRPywHfZu8LvfVn2482T9ZqA1_zW1A4,17430
+autogluon/tabular/models/mitra/mitra_model.py,sha256=6yr6r0bZB_nsoZahSX-w6c9spxBPS2-RHnfqa7IyQyQ,9029
+autogluon/tabular/models/mitra/sklearn_interface.py,sha256=Pc0oTosudoOpxZyaJNOlbtFRD8Bi5QsRIN2LTha9STI,17630
+autogluon/tabular/models/mitra/_internal/__init__.py,sha256=dN2dz1pGMgQTFiSf9oYbyq23iJUxV8QNlOX3qw3KUO4,35
+autogluon/tabular/models/mitra/_internal/config/__init__.py,sha256=Exu_Sx6-K-D5peDQ_TibsjZpqAALs2-9IXfq8hu1mwU,40
 autogluon/tabular/models/mitra/_internal/config/config_pretrain.py,sha256=CeaD96EcDX69LdcLTYGlFmYLdBNINEJXRMWmJ6LbhTg,6038
-autogluon/tabular/models/mitra/_internal/config/config_run.py,sha256=DWDgVSy0hnXaHBzkqxB_azFC9VWxn4F4v8rD2S8f4C0,689
-autogluon/tabular/models/mitra/_internal/config/enums.py,sha256=qUOHLdpoM32DWvnaog8P5sdHetY3d7ypS2NSwNWTRUY,3424
+autogluon/tabular/models/mitra/_internal/config/config_run.py,sha256=guZcX9y-ffx7a-ICn8rhedE0fzAOCgUS-l7-kmt7Uq4,678
+autogluon/tabular/models/mitra/_internal/config/enums.py,sha256=VBVDSMTKlm0GJdIX1H_2PIzOgyCd36x2daTWpFtXlQA,3931
+autogluon/tabular/models/mitra/_internal/core/__init__.py,sha256=hgy4uzJfTQFt9hVlbSrOZU9LSUbLM-uZUnG04f1CUcs,31
 autogluon/tabular/models/mitra/_internal/core/callbacks.py,sha256=xYkJUXiGzLvpWcj6a_wRJUK7f_zgjd1BLA8nH6Hc884,2605
-autogluon/tabular/models/mitra/_internal/core/get_loss.py,sha256=d0ElwQ9iUNFXJiXlZZNfB5_qKCaV2bVBYCW7q2mDBwI,2139
+autogluon/tabular/models/mitra/_internal/core/get_loss.py,sha256=Bpf9V9GcQHBW3JGXCUJ-wjL5WdScfynhG6BAiit9AIE,2244
 autogluon/tabular/models/mitra/_internal/core/get_optimizer.py,sha256=UgGO6lduVZTKZmYAmE207o2Dqs4e3_hyzaoSOQ0iK6A,3412
 autogluon/tabular/models/mitra/_internal/core/get_scheduler.py,sha256=2lzdAxDOYZNq76pmK-FjCOX5MX6cqUSMjqVu8BX9jfY,2238
-autogluon/tabular/models/mitra/_internal/core/prediction_metrics.py,sha256=egun6tOenMYlpvjmu9RRqZeojCy3WVYgaAXiHmtT6R4,4429
-autogluon/tabular/models/mitra/_internal/core/trainer_finetune.py,sha256=jz6VgKCCbfsdnfbgEwxZjO_DvLH_0Nu5s6daFmCPl34,17937
+autogluon/tabular/models/mitra/_internal/core/prediction_metrics.py,sha256=49ofw5xdfhPSQ0Gd7EHAtVAD3e3o7u-IVVF7UoXUbio,4395
+autogluon/tabular/models/mitra/_internal/core/trainer_finetune.py,sha256=_xQoq0Kbgl8c2vpxVNdYsU9m1ejLGIQfiAYOdHKK9T4,17528
+autogluon/tabular/models/mitra/_internal/data/__init__.py,sha256=u4ZTvTQNIHqqxilkVqTmYShI2jFMCOyMdv1GRExvtj0,42
 autogluon/tabular/models/mitra/_internal/data/collator.py,sha256=o2F7ODs_eUnV947lCQTx9RugrANidCdiwnZWtdVNJnE,2300
 autogluon/tabular/models/mitra/_internal/data/dataset_finetune.py,sha256=M2QbXjnb5b4CK5qBthWa7bGvsi8Ox8cz_D0u7tBD4Mo,4232
 autogluon/tabular/models/mitra/_internal/data/dataset_split.py,sha256=xpG62WFjg9NTqukKSJx3byq-SFqhxgpIG4jwIl1YuEc,1929
-autogluon/tabular/models/mitra/_internal/data/preprocessor.py,sha256=99k2eBiTqQD0YwGAs8MM42rCAuSgd_zRoMSl5WCPIQs,14000
+autogluon/tabular/models/mitra/_internal/data/preprocessor.py,sha256=BVP1bJGK_Pdo-CzJ4fmhtsZr2zPMGw4fulxZrD0ELZE,13822
+autogluon/tabular/models/mitra/_internal/models/__init__.py,sha256=K0vh5pyrntXp-o7gWNgQ0ZvDbxgeQuRgb6u8ecdjFhA,45
 autogluon/tabular/models/mitra/_internal/models/base.py,sha256=PKpMPT5OT9JFnmYPnhzFUeZPwdNM1e-k97_gW8GZq0Y,468
 autogluon/tabular/models/mitra/_internal/models/embedding.py,sha256=74O6cGWhUyHxg4-wiQwy4sPeDYQze2ekI9H5mLUtSLg,6223
-autogluon/tabular/models/mitra/_internal/models/tab2d.py,sha256=NatE-htx3fSz4yBGq876i0Jc2AcI_QuoB4yY49H6heM,25799
+autogluon/tabular/models/mitra/_internal/models/tab2d.py,sha256=vs6Fe3sTzk8LwJ2nyDmRaOVdh8oFib5dLd6C092HqjU,25691
+autogluon/tabular/models/mitra/_internal/utils/__init__.py,sha256=0mhykAqjMmcEc8Y2od_DMPMk8f66LZHWM7qFdUrPddU,34
 autogluon/tabular/models/mitra/_internal/utils/set_seed.py,sha256=UnXzYfhmfT_tNAofKtLkKpwB9b6HVf9cpI4mKvoBuNM,340
 autogluon/tabular/models/realmlp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/realmlp/realmlp_model.py,sha256=9GD9iL0R9Z0zfW-26Ay7Agh172AdhiZqQlUr96BlbaU,14215
+autogluon/tabular/models/realmlp/realmlp_model.py,sha256=jlQZ78rBHHYjSPgdIMbyOUwG5y9ADC0hDUG9oxlt7IM,14407
 autogluon/tabular/models/rf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/rf/rf_model.py,sha256=VM4An5U_4whIj-sNvK8m4ImvcqVWqFLUOVwWkxp8o8E,21641
 autogluon/tabular/models/rf/rf_quantile.py,sha256=2S8FE8po9lMnZaeKuVkzOUFOcdil46ZbFqm49OuvNZY,36460
@@ -97,12 +103,12 @@ autogluon/tabular/models/rf/compilers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCe
 autogluon/tabular/models/rf/compilers/native.py,sha256=HhaqQRkVuf9UEEJPsHcdYCmuWBMYtyqRwwB_N2qxG2M,1313
 autogluon/tabular/models/rf/compilers/onnx.py,sha256=pvaZWdl2JJaE2pFU0mFugzhnybePqe0x1-5oLOvogA0,4318
 autogluon/tabular/models/tabicl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabicl/tabicl_model.py,sha256=fby1lsElh0EdVtYBSpEDiUBM57BF20JuROc0Cy0AIBk,5946
+autogluon/tabular/models/tabicl/tabicl_model.py,sha256=QGn4-q6nE9j_TU4drTLV7MmsRDdMqsuqDfRzFFvNYgc,6113
 autogluon/tabular/models/tabm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabm/_tabm_internal.py,sha256=LbIohrZYnXiKbD1ZnXWDJQMBLdQTaL90Fag6fkrF3GI,21093
+autogluon/tabular/models/tabm/_tabm_internal.py,sha256=fRQ-s5PN94kWqf3LRDen7su_fd-d332YKxdms30FoZM,21066
 autogluon/tabular/models/tabm/rtdl_num_embeddings.py,sha256=omDKJT0MjniUPUnk8tSU-brE8dXIjw27BHFbYc2bswQ,30119
-autogluon/tabular/models/tabm/tabm_model.py,sha256=43I8429yTq5U2IDp6ATZB27lyewAW20VzdbPxS-01sA,10115
-autogluon/tabular/models/tabm/tabm_reference.py,sha256=h9FXzyeu6b4vXg9nnM3L2I8dYbcE39USr9C4uMnt4Ek,21788
+autogluon/tabular/models/tabm/tabm_model.py,sha256=5x1HuQx0CJ9SqZ1ZcDor2KOzXhd2G2JPdPQpLC52Odw,10252
+autogluon/tabular/models/tabm/tabm_reference.py,sha256=sZt1LGdifDfJyauVb8wBs9h6lXZJVe0fz0v6oIjXw5A,21908
 autogluon/tabular/models/tabpfnmix/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabpfnmix/tabpfnmix_model.py,sha256=7cLjAfstq6Xb-l2DxBdwtSAIanSJN2sMfKPtijDQwXo,16193
 autogluon/tabular/models/tabpfnmix/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -130,7 +136,7 @@ autogluon/tabular/models/tabpfnmix/_internal/models/foundation/foundation_transf
 autogluon/tabular/models/tabpfnmix/_internal/results/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabpfnmix/_internal/results/prediction_metrics.py,sha256=1tRPHyViSSLJ7BkQJi6wai-PwXJ56od86Dy1WWKWZq4,1743
 autogluon/tabular/models/tabpfnv2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabpfnv2/tabpfnv2_model.py,sha256=DC4t-woswX3mWYRWF6zMfpjEtPZXJ9qgr-mKxWVFs3w,14254
+autogluon/tabular/models/tabpfnv2/tabpfnv2_model.py,sha256=sWuC1aaIlEj_xkUxDOruSGib-YH1eLhxdXiYznQLjbE,14399
 autogluon/tabular/models/tabpfnv2/rfpfn/__init__.py,sha256=yE5XAhGxKEFV0JcelZ_JTQZIWGlVEVUQ9a-lxcH_Esc,585
 autogluon/tabular/models/tabpfnv2/rfpfn/configs.py,sha256=lzBY9kKOeBZACVrtRDPHF4ATs9g1rxyNnIs2CMjE20c,1175
 autogluon/tabular/models/tabpfnv2/rfpfn/scoring_utils.py,sha256=uvHsfvnnMdg4tP3_7zAilktkw7nr65LaqfVKXabXAow,6785
@@ -182,11 +188,11 @@ autogluon/tabular/trainer/model_presets/presets.py,sha256=hoWADaOG576Q_XLV1nY_ju
 autogluon/tabular/trainer/model_presets/presets_distill.py,sha256=MnFC2GJc6RmDBNAGbsO2XMfo3PjR8cUrZoilWW8gTYQ,3295
 autogluon/tabular/tuning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/tuning/feature_pruner.py,sha256=9iNku8gVbYEkjuKlyITPJDicsNkoraaQOlINQq9iZlQ,6877
-autogluon.tabular-1.3.2b20250715.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.3.2b20250715.dist-info/METADATA,sha256=9onrK6_jd80PRQnYWcr0yTXAdGMFXnBAZhsZ8ayXgOc,14646
-autogluon.tabular-1.3.2b20250715.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.3.2b20250715.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.3.2b20250715.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250715.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250715.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.3.2b20250715.dist-info/RECORD,,
+autogluon.tabular-1.3.2b20250716.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.3.2b20250716.dist-info/METADATA,sha256=df9p0ej64-F7a4ocWye3srZiu4HOBqimPfppIPqAqaA,14646
+autogluon.tabular-1.3.2b20250716.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.3.2b20250716.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.3.2b20250716.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250716.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250716.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.3.2b20250716.dist-info/RECORD,,