diff --git a/_mlflow_config_common.ps1 b/_mlflow_config_common.ps1 new file mode 100644 index 0000000..8efe19e --- /dev/null +++ b/_mlflow_config_common.ps1 @@ -0,0 +1,2 @@ +$BACKEND_STORE_DB_PATH = "./mlflow/mlruns.sqlite" +$BACKEND_URI = "sqlite:///$BACKEND_STORE_DB_PATH" diff --git a/_mlflow_config_common.sh b/_mlflow_config_common.sh new file mode 100644 index 0000000..68634ac --- /dev/null +++ b/_mlflow_config_common.sh @@ -0,0 +1,4 @@ +set -eu + +BACKEND_STORE_DB_PATH="${BACKEND_STORE_DB_PATH:-./mlflow/mlruns.sqlite}" +BACKEND_URI="sqlite:///$BACKEND_STORE_DB_PATH" diff --git a/gc_mlflow.ps1 b/gc_mlflow.ps1 new file mode 100644 index 0000000..b040ff0 --- /dev/null +++ b/gc_mlflow.ps1 @@ -0,0 +1,5 @@ +$ErrorActionPreference = "Stop" + +. $PSScriptRoot\_mlflow_config_common.ps1 + +& mlflow gc --backend-store-uri="$BACKEND_URI" @args diff --git a/gc_mlflow.sh b/gc_mlflow.sh new file mode 100644 index 0000000..bac4862 --- /dev/null +++ b/gc_mlflow.sh @@ -0,0 +1,7 @@ +#!/bin/sh + +set -eu + +. _mlflow_config_common.sh + +& mlflow gc --backend-store-uri="$BACKEND_URI" "$@" diff --git a/run_mlflow_server.ps1 b/run_mlflow_server.ps1 index e050933..014edbb 100644 --- a/run_mlflow_server.ps1 +++ b/run_mlflow_server.ps1 @@ -1,12 +1,11 @@ $ErrorActionPreference = "Stop" -$BACKEND_STORE_DB_PATH = "./mlflow/mlruns.sqlite" +. $PSScriptRoot\_mlflow_config_common.ps1 + $DEFAULT_ARTIFACT_ROOT = "./mlflow/mlartifacts/" $MLFLOW_PORT = if ($env:MLFLOW_PORT) { $env:MLFLOW_PORT } else { 5000 } -$BACKEND_URI = "sqlite:///$BACKEND_STORE_DB_PATH" - New-Item -ItemType Directory -Force -Path $DEFAULT_ARTIFACTS_ROOT | Out-Null & mlflow server ` diff --git a/run_mlflow_server.sh b/run_mlflow_server.sh index 1d20505..b9251ef 100755 --- a/run_mlflow_server.sh +++ b/run_mlflow_server.sh @@ -2,11 +2,11 @@ set -eu -DEFAULT_ARTIFACTS_ROOT="./mlflow/mlartifacts/" +. _mlflow_config_common.sh -: "${MLFLOW_PORT:=5000}" +DEFAULT_ARTIFACT_ROOT="./mlflow/mlartifacts/" -BACKEND_URI="sqlite:///$BACKEND_STORE_DB_PATH" +: "${MLFLOW_PORT:=5000}" mkdir -p "${DEFAULT_ARTIFACT_ROOT}"