fix type-errors by declaring the type of o_dimensions in convert_dimensions_to_optuna_space
This commit is contained in:
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user