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

autogluon/tabular/models/mitra/_internal/config/config_run.py

@@ -29,4 +29,4 @@ class ConfigRun(ConfigSaveLoadMixin):
             seed=seed,
             model_name=model_name,
             hyperparams=hyperparams
-        )
+        )

autogluon/tabular/models/mitra/_internal/config/enums.py

@@ -159,4 +159,4 @@ class MetricName(StrEnum):
 class LossName(StrEnum):
     CROSS_ENTROPY = "cross_entropy"
     MSE = "mse"
-    MAE = "mae"
+    MAE = "mae"

autogluon/tabular/models/mitra/_internal/core/get_loss.py

@@ -51,4 +51,4 @@ def get_loss_pretrain(cfg: ConfigPretrain):
     elif cfg.data.task == Task.CLASSIFICATION:
         return CrossEntropyLossExtraBatch(cfg.optim.label_smoothing)
     else:
-        raise ValueError(f"Unsupported task {cfg.data.task} and (regression) loss {cfg.optim.regression_loss}")
+        raise ValueError(f"Unsupported task {cfg.data.task} and (regression) loss {cfg.optim.regression_loss}")

autogluon/tabular/models/mitra/_internal/core/prediction_metrics.py

@@ -129,4 +129,4 @@ class PredictionMetricsTracker():
         y_pred = np.concatenate(self.ys_pred, axis=0)
         y_true = np.concatenate(self.ys_true, axis=0)
 
-        return PredictionMetrics.from_prediction(y_pred, y_true, self.task)
+        return PredictionMetrics.from_prediction(y_pred, y_true, self.task)

autogluon/tabular/models/mitra/_internal/core/trainer_finetune.py

@@ -358,4 +358,4 @@ class TrainerFinetune(BaseEstimator):
                     f"| Train acc: {metrics_train.metrics[MetricName.ACCURACY]:.4f} "
                     f"| Val CE: {metrics_valid.metrics[MetricName.LOG_LOSS]:.4f} "
                     f"| Val acc: {metrics_valid.metrics[MetricName.ACCURACY]:.4f}"
-                ))
+                ))

autogluon/tabular/models/mitra/_internal/data/preprocessor.py

@@ -417,4 +417,4 @@ class Preprocessor():
 
         x = x[:, self.new_feature_order]
 
-        return x
+        return x

autogluon/tabular/models/mitra/_internal/models/tab2d.py

@@ -661,4 +661,4 @@ class MultiheadAttention(torch.nn.Module):
 
         output = self.o(output)
 
-        return output
+        return output

autogluon/tabular/models/mitra/mitra_model.py

@@ -1,8 +1,14 @@
+# TODO: To ensure deterministic operations we need to set torch.use_deterministic_algorithms(True) 
+# and os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'. The CUBLAS environment variable configures
+# the workspace size for certain CUBLAS operations to ensure reproducibility when using CUDA >= 10.2.
+# Both settings are required to ensure deterministic behavior in operations such as matrix multiplications.
+import os
+os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
+
 import os
 from typing import List, Optional
 
 import pandas as pd
-import torch
 
 from autogluon.common.utils.resource_utils import ResourceManager
 from autogluon.core.models import AbstractModel
@@ -41,6 +47,17 @@ class MitraModel(AbstractModel):
         num_cpus: int = 1,
         **kwargs,
     ):
+        
+        # TODO: Reset the number of threads based on the specified num_cpus
+        need_to_reset_torch_threads = False
+        torch_threads_og = None
+        if num_cpus is not None and isinstance(num_cpus, (int, float)):
+            torch_threads_og = torch.get_num_threads()
+            if torch_threads_og != num_cpus:
+                # reset torch threads back to original value after fit
+                torch.set_num_threads(num_cpus)
+                need_to_reset_torch_threads = True
+
         model_cls = self.get_model_cls()
 
         hyp = self._get_model_params()
@@ -69,6 +86,9 @@ class MitraModel(AbstractModel):
             time_limit=time_limit,
         )
 
+        if need_to_reset_torch_threads:
+            torch.set_num_threads(torch_threads_og)
+
     def _set_default_params(self):
         default_params = {
             "device": "cpu",
@@ -142,13 +162,9 @@ class MitraModel(AbstractModel):
     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)
-        
-        # Only request GPU if CUDA is available
-        if torch.cuda.is_available():
-            num_gpus = 1
-        else:
-            num_gpus = 0
-            
+
+        num_gpus = min(1, ResourceManager.get_gpu_count_torch(cuda_only=True))
+
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:
@@ -233,7 +249,7 @@ class MitraModel(AbstractModel):
         **kwargs,
     ) -> int:
         rows, features = X.shape[0], X.shape[1]
-        
+
         # For very small datasets, use a more conservative estimate
         if rows * features < 100:  # Small dataset threshold
             # Use a simpler linear formula for small datasets

autogluon/tabular/models/mitra/sklearn_interface.py

@@ -1,5 +1,6 @@
 import time
 from pathlib import Path
+import contextlib
 
 import numpy as np
 import pandas as pd
@@ -311,23 +312,25 @@ class MitraClassifier(MitraBase, ClassifierMixin):
             Returns self
         """
 
-        if isinstance(X, pd.DataFrame):
-            X = X.values
-        if isinstance(y, pd.Series):
-            y = y.values
+        with mitra_deterministic_context():
 
-        self.X, self.y = X, y
+            if isinstance(X, pd.DataFrame):
+                X = X.values
+            if isinstance(y, pd.Series):
+                y = y.values
 
-        if X_val is not None and y_val is not None:
-            if isinstance(X_val, pd.DataFrame):
-                X_val = X_val.values
-            if isinstance(y_val, pd.Series):
-                y_val = y_val.values
-            X_train, X_valid, y_train, y_valid = X, X_val, y, y_val
-        else:
-            X_train, X_valid, y_train, y_valid = self._split_data(X, y)
+            self.X, self.y = X, y
 
-        return self._train_ensemble(X_train, y_train, X_valid, y_valid, self.task, DEFAULT_CLASSES, n_classes=DEFAULT_CLASSES, time_limit=time_limit)
+            if X_val is not None and y_val is not None:
+                if isinstance(X_val, pd.DataFrame):
+                    X_val = X_val.values
+                if isinstance(y_val, pd.Series):
+                    y_val = y_val.values
+                X_train, X_valid, y_train, y_valid = X, X_val, y, y_val
+            else:
+                X_train, X_valid, y_train, y_valid = self._split_data(X, y)
+
+            return self._train_ensemble(X_train, y_train, X_valid, y_valid, self.task, DEFAULT_CLASSES, n_classes=DEFAULT_CLASSES, time_limit=time_limit)
 
     def predict(self, X):
         """
@@ -363,15 +366,19 @@ class MitraClassifier(MitraBase, ClassifierMixin):
         p : ndarray of shape (n_samples, n_classes)
             The class probabilities of the input samples
         """
-        if isinstance(X, pd.DataFrame):
-            X = X.values
 
-        preds = []
-        for trainer in self.trainers:
-            logits = trainer.predict(self.X, self.y, X)[...,:len(np.unique(self.y))] # Remove extra classes
-            preds.append(np.exp(logits) / np.exp(logits).sum(axis=1, keepdims=True)) # Softmax
-        preds = sum(preds) / len(preds)  # Averaging ensemble predictions
-        return preds
+        with mitra_deterministic_context():
+
+            if isinstance(X, pd.DataFrame):
+                X = X.values
+
+            preds = []
+            for trainer in self.trainers:
+                logits = trainer.predict(self.X, self.y, X)[...,:len(np.unique(self.y))] # Remove extra classes
+                preds.append(np.exp(logits) / np.exp(logits).sum(axis=1, keepdims=True)) # Softmax
+            preds = sum(preds) / len(preds)  # Averaging ensemble predictions
+
+            return preds
 
 
 class MitraRegressor(MitraBase, RegressorMixin):
@@ -433,23 +440,25 @@ class MitraRegressor(MitraBase, RegressorMixin):
             Returns self
         """
 
-        if isinstance(X, pd.DataFrame):
-            X = X.values
-        if isinstance(y, pd.Series):
-            y = y.values
+        with mitra_deterministic_context():
 
-        self.X, self.y = X, y
+            if isinstance(X, pd.DataFrame):
+                X = X.values
+            if isinstance(y, pd.Series):
+                y = y.values
 
-        if X_val is not None and y_val is not None:
-            if isinstance(X_val, pd.DataFrame):
-                X_val = X_val.values
-            if isinstance(y_val, pd.Series):
-                y_val = y_val.values
-            X_train, X_valid, y_train, y_valid = X, X_val, y, y_val
-        else:
-            X_train, X_valid, y_train, y_valid = self._split_data(X, y)
+            self.X, self.y = X, y
+
+            if X_val is not None and y_val is not None:
+                if isinstance(X_val, pd.DataFrame):
+                    X_val = X_val.values
+                if isinstance(y_val, pd.Series):
+                    y_val = y_val.values
+                X_train, X_valid, y_train, y_valid = X, X_val, y, y_val
+            else:
+                X_train, X_valid, y_train, y_valid = self._split_data(X, y)
 
-        return self._train_ensemble(X_train, y_train, X_valid, y_valid, self.task, 1, time_limit=time_limit)
+            return self._train_ensemble(X_train, y_train, X_valid, y_valid, self.task, 1, time_limit=time_limit)
 
     def predict(self, X):
         """
@@ -465,8 +474,30 @@ class MitraRegressor(MitraBase, RegressorMixin):
         y : ndarray of shape (n_samples,)
             The predicted values
         """
-        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
+        with mitra_deterministic_context():
+
+            if isinstance(X, pd.DataFrame):
+                X = X.values
+            
+            preds = []
+            for trainer in self.trainers:
+                preds.append(trainer.predict(self.X, self.y, X))
+        
+            return sum(preds) / len(preds)  # Averaging ensemble predictions
+    
+
+@contextlib.contextmanager
+def mitra_deterministic_context():
+    """Context manager to set deterministic settings only for Mitra operations."""
+    
+    original_deterministic_algorithms_set = False
+
+    try:
+        torch.use_deterministic_algorithms(True)
+        original_deterministic_algorithms_set = True
+        yield
+        
+    finally:
+        if original_deterministic_algorithms_set:
+            torch.use_deterministic_algorithms(False)

autogluon/tabular/models/realmlp/realmlp_model.py

@@ -274,9 +274,9 @@ class RealMLPModel(AbstractModel):
     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)
-        # Only request GPU if CUDA is available (RealMLP doesn't support MPS)
-        import torch
-        num_gpus = 1 if torch.cuda.is_available() else 0
+
+        num_gpus = min(1, ResourceManager.get_gpu_count_torch(cuda_only=True))
+
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:

autogluon/tabular/models/tabicl/tabicl_model.py

@@ -111,9 +111,8 @@ class TabICLModel(AbstractModel):
     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)
-        # Only request GPU if CUDA is available (TabICL doesn't support MPS)
-        import torch
-        num_gpus = 1 if torch.cuda.is_available() else 0
+
+        num_gpus = min(1, ResourceManager.get_gpu_count_torch(cuda_only=True))
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:

autogluon/tabular/models/tabm/tabm_model.py

@@ -150,9 +150,8 @@ class TabMModel(AbstractModel):
     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)
-        # 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
+
+        num_gpus = min(1, ResourceManager.get_gpu_count_torch(cuda_only=True))
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:

autogluon/tabular/models/tabpfnv2/tabpfnv2_model.py

@@ -282,9 +282,9 @@ class TabPFNV2Model(AbstractModel):
     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)
-        # 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
+
+        num_gpus = min(1, ResourceManager.get_gpu_count_torch(cuda_only=True))
+
         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.2b20250716"
+__version__ = "1.3.2b20250718"
 __lite__ = False

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.3.2b20250716
+Version: 1.3.2b20250718
 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.2b20250716
-Requires-Dist: autogluon.features==1.3.2b20250716
+Requires-Dist: autogluon.core==1.3.2b20250718
+Requires-Dist: autogluon.features==1.3.2b20250718
 Provides-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: einops<0.9,>=0.7; extra == "all"
 Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
 Requires-Dist: torch<2.8,>=2.2; extra == "all"
+Requires-Dist: autogluon.core[all]==1.3.2b20250718; 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: pytabkit<1.6,>=1.5; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
-Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; 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.2b20250716; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.3.2b20250718; extra == "ray"
 Provides-Extra: realmlp
 Requires-Dist: pytabkit<1.6,>=1.5; extra == "realmlp"
 Provides-Extra: skex

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

@@ -1,6 +1,6 @@
-autogluon.tabular-1.3.2b20250716-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.3.2b20250718-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=bPJmn8IYFHHx5wv_RbAGbiMuX0PyNDOc2m8l8Fn-Gx8,91
+autogluon/tabular/version.py,sha256=t7hPQFF0BzYTBfD-vM9hoER3q-C5x0pjSWoVO1dcT0w,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,33 +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=6yr6r0bZB_nsoZahSX-w6c9spxBPS2-RHnfqa7IyQyQ,9029
-autogluon/tabular/models/mitra/sklearn_interface.py,sha256=Pc0oTosudoOpxZyaJNOlbtFRD8Bi5QsRIN2LTha9STI,17630
+autogluon/tabular/models/mitra/mitra_model.py,sha256=XiTlzy-RbbHe1t8VCU1y976zmqrDAkO_HhkBiSlk7mM,9985
+autogluon/tabular/models/mitra/sklearn_interface.py,sha256=nX830-_7KYjMnwJ8m8jhCfG7BXU379Ecn5Lu3RvN8Us,18513
 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=guZcX9y-ffx7a-ICn8rhedE0fzAOCgUS-l7-kmt7Uq4,678
-autogluon/tabular/models/mitra/_internal/config/enums.py,sha256=VBVDSMTKlm0GJdIX1H_2PIzOgyCd36x2daTWpFtXlQA,3931
+autogluon/tabular/models/mitra/_internal/config/config_run.py,sha256=CVna6KOwmF-rIxcyH3mHm63jvM1C6RdFbRLgUGEXDn0,677
+autogluon/tabular/models/mitra/_internal/config/enums.py,sha256=hlyhgXHvHZKgYK1z3DHSHxEsuCHOE7Y2AdokjOG8SWs,3930
 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=Bpf9V9GcQHBW3JGXCUJ-wjL5WdScfynhG6BAiit9AIE,2244
+autogluon/tabular/models/mitra/_internal/core/get_loss.py,sha256=hv0t7zvyZ-DgA5PbKpbX_ayq8tEvuW_nJhbudMDqkDk,2243
 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=49ofw5xdfhPSQ0Gd7EHAtVAD3e3o7u-IVVF7UoXUbio,4395
-autogluon/tabular/models/mitra/_internal/core/trainer_finetune.py,sha256=_xQoq0Kbgl8c2vpxVNdYsU9m1ejLGIQfiAYOdHKK9T4,17528
+autogluon/tabular/models/mitra/_internal/core/prediction_metrics.py,sha256=fai0VnDm0mNjJzx8e1JXdB77PKQsmfbtn8zybD9_qD0,4394
+autogluon/tabular/models/mitra/_internal/core/trainer_finetune.py,sha256=LWw60of990QFYKAmKZJytERjj5_m1sveYyRFqPcb6DE,17527
 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=BVP1bJGK_Pdo-CzJ4fmhtsZr2zPMGw4fulxZrD0ELZE,13822
+autogluon/tabular/models/mitra/_internal/data/preprocessor.py,sha256=zx2pWrpDaGSSawPaj7ieRjFOtct_Fyh8LYjo_YtlNG0,13821
 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=vs6Fe3sTzk8LwJ2nyDmRaOVdh8oFib5dLd6C092HqjU,25691
+autogluon/tabular/models/mitra/_internal/models/tab2d.py,sha256=w73QQrXZA7m2fdEPpJDVx-XVZK8xWdc_Q1F38uAZiZA,25690
 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=jlQZ78rBHHYjSPgdIMbyOUwG5y9ADC0hDUG9oxlt7IM,14407
+autogluon/tabular/models/realmlp/realmlp_model.py,sha256=hS3n6spbhZ2bTXqP4t73UnzrSNiUqiaQqPakNQHrS9Y,14332
 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
@@ -103,11 +103,11 @@ 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=QGn4-q6nE9j_TU4drTLV7MmsRDdMqsuqDfRzFFvNYgc,6113
+autogluon/tabular/models/tabicl/tabicl_model.py,sha256=bOCOW2E2bcWQRik2gmebKDEzevswQO_3WAF0JVX-Sis,6038
 autogluon/tabular/models/tabm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=5x1HuQx0CJ9SqZ1ZcDor2KOzXhd2G2JPdPQpLC52Odw,10252
+autogluon/tabular/models/tabm/tabm_model.py,sha256=IQ4RHM1wnf9GHuEa1zDO_yWUPfmh5xUMEVtQ4EFeQRI,10152
 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
@@ -136,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=sWuC1aaIlEj_xkUxDOruSGib-YH1eLhxdXiYznQLjbE,14399
+autogluon/tabular/models/tabpfnv2/tabpfnv2_model.py,sha256=dqjDUpIKQ-SIvbeaDVTq1LfmH4iJ1qRVKpb5_ZMM6oE,14296
 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
@@ -188,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.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,,
+autogluon.tabular-1.3.2b20250718.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.3.2b20250718.dist-info/METADATA,sha256=edzh0r-bATMf-HfQBg7-Gdox3Bq8KfuT101utiZe35s,14646
+autogluon.tabular-1.3.2b20250718.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.3.2b20250718.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.3.2b20250718.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250718.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250718.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.3.2b20250718.dist-info/RECORD,,