Coverage for source/model/model_blue_prints/mock_blue_print.py: 86%

1# model/model_blue_prints/mock_blue_print.py 

2 

3# global imports 

4 

5# local imports 

6from source.model import BluePrintBase 

7from source.model import ModelAdapterBase 

8 

9class MockBluePrint(BluePrintBase): 

10 """""" 

11 

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

13 """""" 

14 

15 self.__model_adapter_to_be_returned: ModelAdapterBase = model_to_be_returned 

16 

17 def instantiate_model(self, input_shape: tuple[int, int], output_length: int, 

18 spatial_data_shape: tuple[int, int], **kwargs) -> ModelAdapterBase: 

19 """ 

20 Returns the pre-configured model regardless of input parameters. 

21 

22 This method implements the abstract method from BluePrintBase but 

23 ignores the input parameters and simply returns the model provided 

24 at initialization. 

25 

26 Parameters: 

27 **kwargs: Variable keyword arguments (ignored). 

28 

29 Returns: 

30 ModelAdapterBase: The pre-configured model provided at initialization. 

31 """ 

32 

33 return self.__model_adapter_to_be_returned