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

View File

@@ -54,6 +54,7 @@ optuna_samplers_dict = {
"QMCSampler": optuna.samplers.QMCSampler
}
class HyperOptimizer:
"""
HyperoptOptimizer class
@@ -396,7 +397,7 @@ class HyperOptimizer:
}
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:
if isinstance(original_dim, SKDecimal):
o_dimensions[original_dim.name] = optuna.distributions.FloatDistribution(