Coverage for source/agent/agents/agent_base.py: 64%

1# agent/agent_base.py 

2 

3# global imports 

4from typing import Optional, Callable 

5from source.model import ModelAdapterBase 

6 

7# local imports 

8 

9class AgentBase(): 

10 """""" 

11 

12 def __init__(self, model_adapter: ModelAdapterBase) -> None: 

13 """""" 

14 

15 self._model_adapter: ModelAdapterBase = model_adapter 

16 

17 def load_model(self, model_path: str) -> None: 

18 """""" 

19 

20 self._model_adapter.load_model(model_path) 

21 

22 def save_model(self, model_path: str) -> None: 

23 """""" 

24 

25 self._model_adapter.save_model(model_path) 

26 

27 def print_summary(self, print_function: Optional[Callable] = print) -> None: 

28 """""" 

29 

30 self._model_adapter.print_summary(print_function)