adapt build_method

Author: AnFreTh

mambular/models/sklearn_base_regressor.py

@@ -181,7 +181,7 @@ def build_model(
             y_val=y_val,
             val_size=val_size,
             random_state=random_state,
-            regression=False,
+            regression=True,
             **dataloader_kwargs
         )
 
@@ -308,7 +308,7 @@ def fit(
         self : object
             The fitted regressor.
         """
-        if not self.built and not rebuild:
+        if rebuild:
             if not isinstance(X, pd.DataFrame):
                 X = pd.DataFrame(X)
             if isinstance(y, pd.Series):
@@ -346,6 +346,9 @@ def fit(
                 weight_decay=weight_decay,
             )
 
+        else:
+            assert self.built, "The model must be built before calling the fit method."
+
         early_stop_callback = EarlyStopping(
             monitor=monitor, min_delta=0.00, patience=patience, verbose=False, mode=mode
         )