fix some formatting issues
This commit is contained in:
@@ -147,9 +147,7 @@ class Hyperopt:
|
|||||||
self.print_all,
|
self.print_all,
|
||||||
)
|
)
|
||||||
|
|
||||||
def run_optimizer_parallel(
|
def run_optimizer_parallel(self, parallel: Parallel, asked: list[list]) -> list[dict[str, Any]]:
|
||||||
self, parallel: Parallel, asked: list[list]
|
|
||||||
) -> list[dict[str, Any]]:
|
|
||||||
"""Start optimizer in a parallel way"""
|
"""Start optimizer in a parallel way"""
|
||||||
|
|
||||||
def optimizer_wrapper(*args, **kwargs):
|
def optimizer_wrapper(*args, **kwargs):
|
||||||
@@ -160,9 +158,7 @@ class Hyperopt:
|
|||||||
|
|
||||||
return self.hyperopter.generate_optimizer(*args, **kwargs)
|
return self.hyperopter.generate_optimizer(*args, **kwargs)
|
||||||
|
|
||||||
return parallel(
|
return parallel(delayed(wrap_non_picklable_objects(optimizer_wrapper))(v) for v in asked)
|
||||||
delayed(wrap_non_picklable_objects(optimizer_wrapper))(v) for v in asked
|
|
||||||
)
|
|
||||||
|
|
||||||
def _set_random_state(self, random_state: int | None) -> int:
|
def _set_random_state(self, random_state: int | None) -> int:
|
||||||
return random_state or random.randint(1, 2**16 - 1) # noqa: S311
|
return random_state or random.randint(1, 2**16 - 1) # noqa: S311
|
||||||
@@ -287,9 +283,7 @@ class Hyperopt:
|
|||||||
asked, is_random = self.get_asked_points(
|
asked, is_random = self.get_asked_points(
|
||||||
n_points=1, dimensions=self.hyperopter.o_dimensions
|
n_points=1, dimensions=self.hyperopter.o_dimensions
|
||||||
)
|
)
|
||||||
f_val0 = self.hyperopter.generate_optimizer(
|
f_val0 = self.hyperopter.generate_optimizer(asked[0].params)
|
||||||
asked[0].params
|
|
||||||
)
|
|
||||||
self.opt.tell(asked[0], [f_val0["loss"]])
|
self.opt.tell(asked[0], [f_val0["loss"]])
|
||||||
self.evaluate_result(f_val0, 1, is_random[0])
|
self.evaluate_result(f_val0, 1, is_random[0])
|
||||||
pbar.update(task, advance=1)
|
pbar.update(task, advance=1)
|
||||||
@@ -308,7 +302,6 @@ class Hyperopt:
|
|||||||
|
|
||||||
f_val = self.run_optimizer_parallel(
|
f_val = self.run_optimizer_parallel(
|
||||||
parallel,
|
parallel,
|
||||||
# self.hyperopter.backtesting,
|
|
||||||
[asked1.params for asked1 in asked],
|
[asked1.params for asked1 in asked],
|
||||||
)
|
)
|
||||||
f_val_loss = [v["loss"] for v in f_val]
|
f_val_loss = [v["loss"] for v in f_val]
|
||||||
|
|||||||
@@ -605,9 +605,7 @@ def test_generate_optimizer(mocker, hyperopt_conf) -> None:
|
|||||||
hyperopt.hyperopter.min_date = dt_utc(2017, 12, 10)
|
hyperopt.hyperopter.min_date = dt_utc(2017, 12, 10)
|
||||||
hyperopt.hyperopter.max_date = dt_utc(2017, 12, 13)
|
hyperopt.hyperopter.max_date = dt_utc(2017, 12, 13)
|
||||||
hyperopt.hyperopter.init_spaces()
|
hyperopt.hyperopter.init_spaces()
|
||||||
generate_optimizer_value = hyperopt.hyperopter.generate_optimizer(
|
generate_optimizer_value = hyperopt.hyperopter.generate_optimizer(optimizer_param)
|
||||||
optimizer_param
|
|
||||||
)
|
|
||||||
assert generate_optimizer_value == response_expected
|
assert generate_optimizer_value == response_expected
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user