From 65a641e1014c989275fb5958a58018758d0116c0 Mon Sep 17 00:00:00 2001 From: Albert Nieto Date: Fri, 27 Sep 2024 15:07:06 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=91=20Fixed=20different=20combination?= =?UTF-8?q?=20length=20in=20grid=20search?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- qcml/bench/grid_search.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/qcml/bench/grid_search.py b/qcml/bench/grid_search.py index 641de74..209e6b7 100644 --- a/qcml/bench/grid_search.py +++ b/qcml/bench/grid_search.py @@ -195,7 +195,17 @@ def _evaluate_combinations( self.results = [] total_combinations = len(self.combinations) - for idx, (params, trans_func, trans_params) in enumerate(self.combinations): + for idx, combination in enumerate(self.combinations): + if len(combination) == 1: + params = combination[0] + trans_func = None + trans_params = None + elif len(combination) == 3: + params, trans_func, trans_params = combination + else: + logger.error(f"Invalid combination format: {combination}") + continue # Skip invalid combinations + try: evaluation_result = self.evaluator.evaluate( classifier,