def _create_project(self, project_name, with_repo=False): proj = mlrun.new_project(project_name, str(self.assets_path)) proj.set_function( "prep_data.py", "prep-data", image="mlrun/mlrun", handler="prep_data", with_repo=with_repo, ) proj.set_function("hub://describe") proj.set_function("hub://sklearn_classifier", "train") proj.set_function("hub://test_classifier", "test") proj.set_function("hub://v2_model_server", "serving") proj.set_artifact("data", Artifact(target_path=data_url)) proj.spec.params = {"label_column": "label"} arg = EntrypointParam( "model_pkg_class", type="str", default=model_pkg_class, doc="model package/algorithm", ) proj.set_workflow("main", "./kflow.py", args_schema=[arg]) proj.set_workflow("newflow", "./newflow.py", handler="newpipe") proj.save() return proj
def test_artifact(create_server): server: Server = create_server() db = server.conn prj, uid, key, body = 'p7', 'u199', 'k800', 'cucumber' artifact = Artifact(key, body) db.store_artifact(key, artifact, uid, project=prj)
def test_artifact(create_server): server: Server = create_server() db = server.conn prj, uid, key, body = "p7", "u199", "k800", "cucumber" artifact = Artifact(key, body) db.store_artifact(key, artifact, uid, project=prj)
def test_artifacts(create_server): server: Server = create_server() db = server.conn prj, uid, key, body = 'p9', 'u19', 'k802', 'tomato' artifact = Artifact(key, body) db.store_artifact(key, artifact, uid, project=prj) artifacts = db.list_artifacts(project=prj, tag='*') assert len(artifacts) == 1, 'bad number of artifacs' db.del_artifacts(project=prj, tag='*') artifacts = db.list_artifacts(project=prj, tag='*') assert len(artifacts) == 0, 'bad number of artifacs after del'
def test_artifacts(create_server): server: Server = create_server() db = server.conn prj, uid, key, body = "p9", "u19", "k802", "tomato" artifact = Artifact(key, body) db.store_artifact(key, artifact, uid, project=prj) artifacts = db.list_artifacts(project=prj, tag="*") assert len(artifacts) == 1, "bad number of artifacs" db.del_artifacts(project=prj, tag="*") artifacts = db.list_artifacts(project=prj, tag="*") assert len(artifacts) == 0, "bad number of artifacs after del"
def test_artifacts(create_server): server: Server = create_server() db = server.conn prj, uid, key, body = "p9", "u19", "k802", "tomato" artifact = Artifact(key, body) db.store_artifact(key, artifact, uid, project=prj) db.store_artifact(key, artifact, uid, project=prj, iter=42) artifacts = db.list_artifacts(project=prj, tag="*") assert len(artifacts) == 2, "bad number of artifacts" artifacts = db.list_artifacts(project=prj, tag="*", iter=0) assert len(artifacts) == 1, "bad number of artifacts" # Only 1 will be returned since it's only looking for iter 0 artifacts = db.list_artifacts(project=prj, tag="*", best_iteration=True) assert len(artifacts) == 1, "bad number of artifacts" db.del_artifacts(project=prj, tag="*") artifacts = db.list_artifacts(project=prj, tag="*") assert len(artifacts) == 0, "bad number of artifacts after del"
def test_artifacts(create_server): server: Server = create_server() db = server.conn prj, uid, key, body = "p9", "u19", "k802", "tomato" artifact = Artifact(key, body, target_path="a.txt") db.store_artifact(key, artifact, uid, project=prj) db.store_artifact(key, artifact, uid, project=prj, iter=42) artifacts = db.list_artifacts(project=prj, tag="*") assert len(artifacts) == 2, "bad number of artifacts" assert artifacts.objects()[0].key == key, "not a valid artifact object" assert artifacts.dataitems()[0].url, "not a valid artifact dataitem" artifacts = db.list_artifacts(project=prj, tag="*", iter=0) assert len(artifacts) == 1, "bad number of artifacts" # Only 1 will be returned since it's only looking for iter 0 artifacts = db.list_artifacts(project=prj, tag="*", best_iteration=True) assert len(artifacts) == 1, "bad number of artifacts" db.del_artifacts(project=prj, tag="*") artifacts = db.list_artifacts(project=prj, tag="*") assert len(artifacts) == 0, "bad number of artifacts after del"