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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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'] = ''