def test_get_model(ebnt: Ebonite):
task = ebnt.get_or_create_task("Project", "Task")
reg, data = train_model()
model = task.create_and_push_model(reg, data, 'mymodel')
assert ebnt.get_model(project='Project', task='Task',
model_name='mymodel') == model
def test_get_model(ebnt: Ebonite, regression_and_data):
task = ebnt.get_or_create_task("Project", "Task")
reg, data = regression_and_data
model = task.create_and_push_model(reg, data, 'mymodel')
get_model = ebnt.get_model(project='Project',
task='Task',
model_name='mymodel')
assert get_model == model
assert get_model.has_meta_repo
assert get_model.has_artifact_repo
def test_create_model(ebnt: Ebonite, regression_and_data):
reg, data = regression_and_data
model = ebnt.create_model(reg, data, 'sklearn_model')
assert isinstance(model, Model)
assert ebnt.get_model(model.name, model.task) == model