Fix some type errors

This commit is contained in:
Matthias 2021-05-29 16:56:36 +02:00
parent 2bf17f71e7
commit 8cdd1e3aef
2 changed files with 10 additions and 6 deletions

View File

@ -136,7 +136,10 @@ def start_hyperopt_show(args: Dict[str, Any]) -> None:
# Export parameters ... # Export parameters ...
# TODO: make this optional? otherwise it'll overwrite previous parameters ... # TODO: make this optional? otherwise it'll overwrite previous parameters ...
fn = HyperoptTools.get_strategy_filename(config, strategy_name) fn = HyperoptTools.get_strategy_filename(config, strategy_name)
HyperoptTools.export_params(val, strategy_name, fn.with_suffix('.json')) if fn:
HyperoptTools.export_params(val, strategy_name, fn.with_suffix('.json'))
else:
logger.warn("Strategy not found, not exporting parameter file.")
HyperoptTools.show_epoch_details(val, total_epochs, print_json, no_header, HyperoptTools.show_epoch_details(val, total_epochs, print_json, no_header,
header_str="Epoch details") header_str="Epoch details")

View File

@ -3,7 +3,7 @@ import io
import logging import logging
from copy import deepcopy from copy import deepcopy
from pathlib import Path from pathlib import Path
from typing import Any, Dict, List from typing import Any, Dict, List, Optional
import rapidjson import rapidjson
import tabulate import tabulate
@ -21,18 +21,19 @@ logger = logging.getLogger(__name__)
class HyperoptTools(): class HyperoptTools():
@staticmethod @staticmethod
def get_strategy_filename(config: Dict, strategy_name: str) -> Path: def get_strategy_filename(config: Dict, strategy_name: str) -> Optional[Path]:
""" """
Get Strategy-location (filename) from strategy_name Get Strategy-location (filename) from strategy_name
""" """
from freqtrade.resolvers.strategy_resolver import StrategyResolver from freqtrade.resolvers.strategy_resolver import StrategyResolver
directory = Path(config.get('strategy_path', config['user_data_dir'] / USERPATH_STRATEGIES)) directory = Path(config.get('strategy_path', config['user_data_dir'] / USERPATH_STRATEGIES))
strategy_objs = StrategyResolver.search_all_objects(directory, False) strategy_objs = StrategyResolver.search_all_objects(directory, False)
strategy = [s for s in strategy_objs if s['name'] == strategy_name] strategies = [s for s in strategy_objs if s['name'] == strategy_name]
if strategy: if strategies:
strategy = strategy[0] strategy = strategies[0]
return Path(strategy['location']) return Path(strategy['location'])
return None
@staticmethod @staticmethod
def export_params(params, strategy_name: str, filename: Path): def export_params(params, strategy_name: str, filename: Path):