from image_prediction.model_loader.database.connector import DatabaseConnector


class DatabaseConnectorMock(DatabaseConnector):
    def __init__(self, store: dict):
        self.store = store

    def get_object(self, identifier):
        return self.store[identifier]