Skip to content

Commit

Permalink
feat(images): Add MLflow
Browse files Browse the repository at this point in the history
Signed-off-by: RJ Sampson <[email protected]>
  • Loading branch information
EyeCantCU committed May 6, 2024
1 parent 4c0e420 commit 7de78f3
Show file tree
Hide file tree
Showing 8 changed files with 184 additions and 0 deletions.
9 changes: 9 additions & 0 deletions generated.tf

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

29 changes: 29 additions & 0 deletions images/mlflow/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
<!--monopod:start-->
# mlflow
| | |
| - | - |
| **OCI Reference** | `cgr.dev/chainguard/mlflow` |


* [View Image in Chainguard Academy](https://edu.chainguard.dev/chainguard/chainguard-images/reference/mlflow/overview/)
* [View Image Catalog](https://console.enforce.dev/images/catalog) for a full list of available tags.
* [Contact Chainguard](https://www.chainguard.dev/chainguard-images) for enterprise support, SLAs, and access to older tags.*

---
<!--monopod:end-->

<!--overview:start-->
A minimal, [Wolfi](https://github.com/wolfi-dev)-based image for MLflow, an open source platform for the machine learning lifecycle.

<!--overview:end-->

<!--getting:start-->
## Download this Image
The image is available on `cgr.dev`:

```
docker pull cgr.dev/chainguard/mlflow:latest
```
<!--getting:end-->

<!--body:start--><!--body:end-->
34 changes: 34 additions & 0 deletions images/mlflow/config/main.tf
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
terraform {
required_providers {
apko = { source = "chainguard-dev/apko" }
}
}

variable "extra_packages" {
description = "Additional packages to install."
type = list(string)
default = ["mlflow"]
}

variable "environment" {
default = {}
}

module "accts" {
source = "../../../tflib/accts"
}

output "config" {
value = jsonencode({
contents = {
packages = var.extra_packages
}
accounts = module.accts.block
environment = merge({
"PATH" : "/usr/share/mlflow/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
}, var.environment)
entrypoint = {
command = "/usr/share/mlflow/bin/python3"
}
})
}
13 changes: 13 additions & 0 deletions images/mlflow/generated.tf

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

38 changes: 38 additions & 0 deletions images/mlflow/main.tf
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
terraform {
required_providers {
oci = { source = "chainguard-dev/oci" }
}
}

variable "target_repository" {
description = "The docker repo into which the image and attestations should be published."
}

module "config" {
source = "./config"
}

module "latest" {
source = "../../tflib/publisher"
name = basename(path.module)
target_repository = var.target_repository
config = module.config.config
build-dev = true
}

module "test" {
source = "./tests"
digest = module.latest.image_ref
}

resource "oci_tag" "latest" {
depends_on = [module.test]
digest_ref = module.latest.image_ref
tag = "latest"
}

resource "oci_tag" "latest-dev" {
depends_on = [module.test]
digest_ref = module.latest.dev_ref
tag = "latest-dev"
}
13 changes: 13 additions & 0 deletions images/mlflow/metadata.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
name: mlflow
image: cgr.dev/chainguard/mlflow
logo: https://storage.googleapis.com/chainguard-academy/logos/mlflow.svg
endoflife: ""
console_summary: ""
short_description: |
A minimal, [Wolfi](https://github.com/wolfi-dev)-based image for MLflow, an open source platform for the machine learning lifecycle.
compatibility_notes: ""
readme_file: README.md
upstream_url: https://mlflow.org/
keywords:
- ai
- python
5 changes: 5 additions & 0 deletions images/mlflow/tests/check-mlflow.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
#!/usr/bin/env bash

set -o errexit -o nounset -o errtrace -o pipefail -x

cwd=$(cd "${0%/*}" && pwd)
43 changes: 43 additions & 0 deletions images/mlflow/tests/main.tf
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
terraform {
required_providers {
oci = { source = "chainguard-dev/oci" }
imagetest = { source = "chainguard-dev/imagetest" }
}
}

variable "digest" {
description = "The image digest to run tests over."
}

data "imagetest_inventory" "this" {}

resource "imagetest_container_volume" "volume" {
name = "scratch-volume"
inventory = data.imagetest_inventory.this
}

resource "imagetest_harness_docker" "this" {
name = "jre"
inventory = data.imagetest_inventory.this

mounts = [{
source = path.module
destination = "/tests"
}]

envs = {
"IMAGE_NAME" : var.digest
}
}

resource "imagetest_feature" "basic" {
name = "Test MLflow"
harness = imagetest_harness_docker.this

steps = [{
name = "Import test"
cmd = <<EOT
docker run --rm $IMAGE_NAME -m mlflow
EOT
}]
}

0 comments on commit 7de78f3

Please sign in to comment.