Add type to "trials" parameter

This commit is contained in:
raphael
2021-09-17 14:10:37 -04:00
committed by GitHub
parent 3a98fb72a4
commit 9525a5b96c

View File

@@ -299,7 +299,7 @@ class HyperoptTools():
f"Objective: {results['loss']:.5f}")
@staticmethod
def prepare_trials_columns(trials, legacy_mode: bool, has_drawdown: bool) -> pd.DataFrame:
def prepare_trials_columns(trials: pd.DataFrame, legacy_mode: bool, has_drawdown: bool) -> pd.DataFrame:
trials['Best'] = ''