Source code for tensortrade.strategies

from .trading_strategy import TradingStrategy
from .stable_baselines_strategy import StableBaselinesTradingStrategy
from .tensorforce_trading_strategy import TensorforceTradingStrategy


_registry = {}


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