fix type-errors by declaring the type of o_dimensions in convert_dimensions_to_optuna_space

This commit is contained in:
viotemp1
2025-03-26 16:55:43 +02:00
parent 553dbccec7
commit 59e52bb601
@@ -54,6 +54,7 @@ optuna_samplers_dict = {
"QMCSampler": optuna.samplers.QMCSampler "QMCSampler": optuna.samplers.QMCSampler
} }
class HyperOptimizer: class HyperOptimizer:
""" """
HyperoptOptimizer class HyperoptOptimizer class
@@ -396,7 +397,7 @@ class HyperOptimizer:
} }
def convert_dimensions_to_optuna_space(self, s_dimensions: list[Dimension]) -> dict: def convert_dimensions_to_optuna_space(self, s_dimensions: list[Dimension]) -> dict:
o_dimensions = {} o_dimensions: dict[str, optuna.distributions.BaseDistribution] = {}
for original_dim in s_dimensions: for original_dim in s_dimensions:
if isinstance(original_dim, SKDecimal): if isinstance(original_dim, SKDecimal):
o_dimensions[original_dim.name] = optuna.distributions.FloatDistribution( o_dimensions[original_dim.name] = optuna.distributions.FloatDistribution(