Skip to content

Commit 1e50a2d

Browse files
committed
refactor: simplify hyperopt-auto
1 parent 7d1c2d3 commit 1e50a2d

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

freqtrade/optimize/hyperopt/hyperopt_auto.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -62,20 +62,18 @@ def _get_func(self, name) -> Callable:
6262
else:
6363
return default_func
6464

65-
def _generate_indicator_space(self, category):
66-
for attr_name, attr in self.strategy.enumerate_parameters(category):
67-
if attr.optimize:
68-
yield attr.get_space(attr_name)
69-
7065
def get_indicator_space(
7166
self, category: Literal["buy", "sell", "enter", "exit", "protection"] | str
7267
) -> list:
7368
"""
7469
Get indicator space for a given space.
7570
:param category: parameter space to get.
7671
"""
77-
# TODO: is this necessary, or can we call "generate_space" directly?
78-
indicator_space = list(self._generate_indicator_space(category))
72+
indicator_space = [
73+
attr.get_space(attr_name)
74+
for attr_name, attr in self.strategy.enumerate_parameters(category)
75+
if attr.optimize
76+
]
7977
if len(indicator_space) > 0:
8078
return indicator_space
8179
else:

0 commit comments

Comments
 (0)