fixed TabM interface error

dholzmueller · dholzmueller · commit 81fbf54f7719 · 2025-07-27T15:58:14.000+02:00
diff --git a/pytabkit/models/sklearn/sklearn_interfaces.py b/pytabkit/models/sklearn/sklearn_interfaces.py
@@ -1870,6 +1870,7 @@ def __init__(self, device: Optional[str] = None, random_state: Optional[Union[in
         self.calibration_method = calibration_method
         self.share_training_batches = share_training_batches
         self.val_metric_name = val_metric_name
+        self.train_metric_name = train_metric_name
 
 
 class TabM_D_Classifier(TabMConstructorMixin, AlgInterfaceClassifier):
