12 lines
296 B
Python
12 lines
296 B
Python
from image_prediction.model_loader.loader import ModelLoader
|
|
|
|
|
|
class ModelLoaderMock(ModelLoader):
|
|
|
|
model = None
|
|
classes = None
|
|
|
|
def load_model(self, identifier):
|
|
assert self.model is not None, "Set the model to be returned first via monkeypatching"
|
|
return self.model
|