Turn guard clause into if-else

emdgroup · Apr 26, 2024 · f4d500b · f4d500b
1 parent a88a5c1
commit f4d500b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/baybe/surrogates/utils.py b/baybe/surrogates/utils.py
@@ -99,10 +99,10 @@ def _fit_new(
         if train_y.numel() == 1 or train_y.std() < std_threshold:
             self._model = MeanPredictionSurrogate()
             self._model._fit(searchspace, train_x, train_y)
-            return
 
         # Regular operation
-        _fit_original(self, searchspace, train_x, train_y)
+        else:
+            _fit_original(self, searchspace, train_x, train_y)
 
     # Replace the methods
     cls._posterior = _posterior_new