Skip to content

Commit

Permalink
ci: add a ML model artifact test case
Browse files Browse the repository at this point in the history
Both as a test and an example.

Inspired by:
    https://github.com/kubeflow/model-registry/blob/main/docs/logical_model.md

Signed-off-by: Ramkumar Chinchani <[email protected]>
  • Loading branch information
rchincha committed Mar 19, 2024
1 parent eec277e commit 8115269
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions test/blackbox/pushpull_authn.bats
Original file line number Diff line number Diff line change
Expand Up @@ -173,3 +173,20 @@ EOF
[ "$status" -eq 0 ]
[ $(echo "${lines[-1]}" | jq '.manifests | length') -eq 0 ]
}

@test "ML artifacts" {
# download model data
curl -v -L0 https://github.com/tarilabs/demo20231212/raw/main/v1.nb20231206162408/mnist.onnx -o ${BATS_FILE_TMPDIR}/mnist.onnx

# upload artifact with required annotations and version
run regctl artifact put --annotation description="used for demo purposes" --annotation model_format_name="onnx" --annotation model_format_name="onnx" --annotation model_format_version="1" --artifact-type "application/vnd.model.type" localhost:${zot_port}/models/my-model-from-gh:v1 -f ${BATS_FILE_TMPDIR}/mnist.onnx

# list artifacts
run regctl artifact list localhost:${zot_port}/models/my-model-from-gh:v1 --format raw-body

# list artifacts of type
run regctl artifact list --filter-artifact-type "application/vnd.model.type" localhost:${zot_port}/models/my-model-from-gh:v1 --format raw-body

# get artifact
run regctl artifact get localhost:${zot_port}/models/my-model-from-gh:v1
}

0 comments on commit 8115269

Please sign in to comment.