Merge pull request #204 from hyperopt/gbr_loss

Fix decorator typo GBR
hyperopt · Nov 20, 2023 · d050420 · d050420
2 parents 840cfc9 + 111279d
commit d050420
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hpsklearn/components/ensemble/_gb.py b/hpsklearn/components/ensemble/_gb.py
@@ -240,7 +240,7 @@ def _name(msg):
           msg="Invalid parameter '%s' with value '%s'. Parameter value must be within [0.0, 1.0]")
 @validate(params=["loss"],
           validation_test=lambda param: not isinstance(param, str) or param in ("squared_error", "absolute_error",
-                                                                                "huber" or "quantile"),
+                                                                                "huber", "quantile"),
           msg="Invalid parameter '%s' with value '%s'. "
               "Choose 'squared_error', 'absolute_error', 'huber' or 'quantile'.")
 def gradient_boosting_regressor(name: str,