artificial_artwork.pretrained_model.model_handler_interface

Complexity

Total Complexity

7

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	PretrainedModelRoutinesCapable.get_layers_dict()	0	1	1
A	PretrainedModelRoutinesCapable.get_weights()	0	1	1
A	ModelHandlerInterface.model_load_exception_text()	0	4	1
A	PretrainedModelRoutinesCapable.get_id()	0	1	1
A	ModelHandlerInterface.model_routines()	0	4	1
A	PretrainedModelRoutinesCapable.load_layers()	0	1	1
A	ModelHandlerInterface.environment_variable()	0	4	1

from abc import ABC, abstractmethod
from typing import Dict, Protocol, Tuple
from numpy.typing import NDArray


class PretrainedModelRoutinesCapable(Protocol):
    def load_layers(self, file_path: str) -> NDArray: ...

    def get_id(self, layer: NDArray) -> str: ...
    
    def get_layers_dict(self, layers: NDArray) -> Dict[str, NDArray]: ...

    def get_weights(self, layer: NDArray) -> Tuple[NDArray, NDArray]: ...


class ModelHandlerInterface(ABC):

    @property
    @abstractmethod
    def environment_variable(self) -> str:
        raise NotImplementedError

    @property
    @abstractmethod
    def model_routines(self) -> PretrainedModelRoutinesCapable:
        raise NotImplementedError

    @property
    @abstractmethod
    def model_load_exception_text(self) -> str:
        raise NotImplementedError