10 lines
270 B
Python
10 lines
270 B
Python
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]
|