From 09a502aba499be0fc690e6e8145755f43bc79704 Mon Sep 17 00:00:00 2001 From: Sergiy Matusevych Date: Wed, 14 Aug 2024 20:09:41 -0700 Subject: [PATCH] add quantization_bins check --- mlos_bench/mlos_bench/optimizers/convert_configspace.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mlos_bench/mlos_bench/optimizers/convert_configspace.py b/mlos_bench/mlos_bench/optimizers/convert_configspace.py index 97a10ac489f..ad7968cf69a 100644 --- a/mlos_bench/mlos_bench/optimizers/convert_configspace.py +++ b/mlos_bench/mlos_bench/optimizers/convert_configspace.py @@ -60,6 +60,9 @@ def _monkey_patch_quantization(hp: NumericalHyperparameter, quantization_bins: i quantization_bins : int Number of bins to quantize the hyperparameter into. """ + if quantization_bins <= 1: + raise ValueError(f"{quantization_bins=} :: must be greater than 1.") + # Temporary workaround to dropped quantization support in ConfigSpace 1.0 # See Also: https://github.com/automl/ConfigSpace/issues/390 if not hasattr(hp, "sample_value_mlos_orig"):