Update prepare_trials_columns() return type

Was returning str, updated to pd.DataFrame
This commit is contained in:
raphael
2021-09-17 11:42:33 -04:00
committed by GitHub
parent 4ce1375bf3
commit 3a98fb72a4

View File

@@ -7,6 +7,7 @@ from pathlib import Path
from typing import Any, Dict, Iterator, List, Optional, Tuple from typing import Any, Dict, Iterator, List, Optional, Tuple
import numpy as np import numpy as np
import pandas as pd
import rapidjson import rapidjson
import tabulate import tabulate
from colorama import Fore, Style from colorama import Fore, Style
@@ -298,7 +299,7 @@ class HyperoptTools():
f"Objective: {results['loss']:.5f}") f"Objective: {results['loss']:.5f}")
@staticmethod @staticmethod
def prepare_trials_columns(trials, legacy_mode: bool, has_drawdown: bool) -> str: def prepare_trials_columns(trials, legacy_mode: bool, has_drawdown: bool) -> pd.DataFrame:
trials['Best'] = '' trials['Best'] = ''