geocrop-platform./k8s/base/mlflow.yaml

47 lines
1.2 KiB
YAML

apiVersion: apps/v1
kind: Deployment
metadata:
name: mlflow
namespace: geocrop
spec:
replicas: 1
selector:
matchLabels:
app: mlflow
template:
metadata:
labels:
app: mlflow
spec:
containers:
- name: mlflow
# Using a version that is known to work with postgres
image: ghcr.io/mlflow/mlflow:v2.12.1
command:
- mlflow
- server
- --host=0.0.0.0
- --port=5000
- --backend-store-uri=postgresql+psycopg2://postgres:$(DB_PASSWORD)@geocrop-db:5432/geocrop_gis
- --default-artifact-root=s3://geocrop-models/mlflow-artifacts
env:
- name: DB_PASSWORD
valueFrom:
secretKeyRef:
name: geocrop-db-secret
key: password
- name: AWS_ACCESS_KEY_ID
valueFrom:
secretKeyRef:
name: geocrop-secrets
key: minio-access-key
- name: AWS_SECRET_ACCESS_KEY
valueFrom:
secretKeyRef:
name: geocrop-secrets
key: minio-secret-key
- name: MLFLOW_S3_ENDPOINT_URL
value: http://minio.geocrop.svc.cluster.local:9000
ports:
- containerPort: 5000