def data_set_manager(file_repository_stub, workspace_repository_stub,
ml_model_repository_stub):
data_set_manager = DataSetManager(get_workspace_stub()._id,
DataSourceStub())
data_set_manager.db_is_set = True
data_set_manager.file_repository = file_repository_stub
data_set_manager.workspace_repository = workspace_repository_stub
data_set_manager.ml_model_repository = ml_model_repository_stub
return data_set_manager
def fetch_samples(user_id: ObjectId, workspace_id: ObjectId) -> List[SampleFromWorkspace]:
workspace_stub = get_workspace_stub()
if user_id is not workspace_stub.user_id or workspace_id is not workspace_stub._id:
# TODO raise error
pass
return [get_sample_from_workspace_stub_1(), get_sample_from_workspace_stub_2()]
def last_modified(user_id: ObjectId, workspace_id: ObjectId) -> int:
workspace_stub = get_workspace_stub()
if user_id is not workspace_stub.user_id or workspace_id is not workspace_stub._id:
# TODO raise error
pass
return 1617981582111
def workspace_stub() -> Workspace:
workspace = get_workspace_stub()
workspace.ml_model_refs = []
return workspace
def workspace_sensors():
return get_workspace_stub().sensors