explicit dtype

This commit is contained in:
Italo
2022-01-25 12:29:55 +00:00
parent f7a5b2cb71
commit 30b27ae736
2 changed files with 4 additions and 4 deletions

View File

@@ -40,7 +40,7 @@ class IHyperOpt(ABC):
IHyperOpt.ticker_interval = str(config['timeframe']) # DEPRECATED
IHyperOpt.timeframe = str(config['timeframe'])
def generate_estimator(self, dimensions, **kwargs) -> EstimatorType:
def generate_estimator(self, dimensions: List[Dimension], **kwargs) -> EstimatorType:
"""
Return base_estimator.
Can be any of "GP", "RF", "ET", "GBRT" or an instance of a class