Coverage for source/model/model_adapters/model_adapter_base.py: 71%

1# model/model_adapters/model_adapter_base.py 

2 

3# global imports 

4from abc import ABC, abstractmethod 

5from typing import Any, Callable 

6 

7class ModelAdapterBase(ABC): 

8 """""" 

9 

10 @abstractmethod 

11 def load_model(self, path: str) -> None: 

12 """""" 

13 

14 pass 

15 

16 @abstractmethod 

17 def save_model(self, path: str) -> None: 

18 """""" 

19 

20 pass 

21 

22 @abstractmethod 

23 def print_summary(self, print_function: Callable = print) -> None: 

24 """""" 

25 

26 pass 

27 

28 @abstractmethod 

29 def fit(self, input_data: Any, output_data: Any, **kwargs) -> Any: 

30 """""" 

31 

32 pass 

33 

34 @abstractmethod 

35 def predict(self, data: Any) -> Any: 

36 """""" 

37 

38 pass 

39 

40 @abstractmethod 

41 def get_model(self) -> Any: 

42 """""" 

43 

44 pass