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