diff --git a/image_prediction/model_loader/loaders/mock.py b/image_prediction/model_loader/loaders/mock.py deleted file mode 100644 index 6269261..0000000 --- a/image_prediction/model_loader/loaders/mock.py +++ /dev/null @@ -1,11 +0,0 @@ -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