Adding MLFlow

.env

@@ -0,0 +1,13 @@
+# Conda
+CONDA_TAG=4.12.0
+
+# MLFlow
+MLFLOW_LABEL=1.30.0
+
+# Psycopg 
+PSYCOPG_LABEL=2.9.3 
+
+# Docker
+CONTAINER_REG_USER=jbris
+CONTAINER_REG_NAME=mlflow-docker
+

.gitignore

@@ -102,7 +102,6 @@ celerybeat.pid
 *.sage.py
 
 # Environments
-.env
 .venv
 env/
 venv/

Dockerfile

@@ -0,0 +1,12 @@
+ARG CONDA_TAG
+
+FROM continuumio/miniconda3:${CONDA_TAG}
+
+RUN mkdir -p /mlflow/mlruns
+
+WORKDIR /mlflow
+
+ARG MLFLOW_LABEL
+ARG PSYCOPG_LABEL
+
+RUN conda install -c conda-forge psycopg2=${PSYCOPG_LABEL} mlflow=${MLFLOW_LABEL} -y --quiet

README.md

@@ -1,2 +1,3 @@
 # mlflow-docker
+
 Run MLFlow through Docker

docker-compose.yaml

@@ -0,0 +1,12 @@
+version: "3.7"
+
+services:
+
+  mlflow:
+    image: "continuumio/miniconda3:${CONDA_TAG}"
+    build: 
+      context: .
+      args:
+        CONDA_TAG: $CONDA_TAG
+        MLFLOW_LABEL: $MLFLOW_LABEL
+        PSYCOPG_LABEL: $PSYCOPG_LABEL

docker_build.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+
+###################################################################
+# Main
+###################################################################
+
+docker compose pull
+docker compose build

docker_push.sh

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+
+. .env
+
+###################################################################
+# Main
+###################################################################
+
+echo $CR_PAT | docker login ghcr.io -u USERNAME --password-stdin
+
+# To find DOCKER_IMAGE_HASH: docker image ls
+docker tag "$DOCKER_IMAGE_HASH" "ghcr.io/${CONTAINER_REG_USER}/${CONTAINER_REG_NAME}:latest"
+docker push "ghcr.io/${CONTAINER_REG_USER}/${CONTAINER_REG_NAME}:latest"