Coverage for source/agent/strategies/learning_strategy_handler_base.py: 100%

1# agent/strategies/learning_strategy_handler_base.py 

2 

3# global imports 

4from typing import Any 

5from tensorflow.keras.callbacks import Callback 

6 

7# local imports 

8from source.environment import TradingEnvironment 

9from source.agent import AgentBase 

10from source.model import BluePrintBase 

11 

12class LearningStrategyHandlerBase: 

13 """""" 

14 

15 def create_agent(self, model_blue_print: BluePrintBase, 

16 trading_environment: TradingEnvironment) -> AgentBase: 

17 """""" 

18 

19 raise NotImplementedError("Subclasses must implement this method.") 

20 

21 def fit(self, agent: AgentBase, nr_of_steps: int, nr_of_episodes: int, 

22 callbacks: list[Callback]) -> tuple[list[str], list[dict[str, Any]]]: 

23 """""" 

24 

25 raise NotImplementedError("Subclasses must implement this method.")