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"
