Source code for tensortrade.rewards

from .reward_strategy import RewardStrategy
from .simple_profit_strategy import SimpleProfitStrategy
from .risk_adjusted_return_strategy import RiskAdjustedReturnStrategy

_registry = {
    'simple': SimpleProfitStrategy(),
    'risk-adjusted': RiskAdjustedReturnStrategy()
}


[docs]def get(identifier: str) -> RewardStrategy: """Gets the `RewardStrategy` that matches with the identifier. Arguments: identifier: The identifier for the `RewardStrategy` Raises: KeyError: if identifier is not associated with any `RewardStrategy` """ if identifier not in _registry.keys(): raise KeyError( 'Identifier {} is not associated with any `RewardStrategy`.'.format(identifier)) return _registry[identifier]