diff --git a/run_mlflow_server.ps1 b/run_mlflow_server.ps1 new file mode 100644 index 0000000..e050933 --- /dev/null +++ b/run_mlflow_server.ps1 @@ -0,0 +1,15 @@ +$ErrorActionPreference = "Stop" + +$BACKEND_STORE_DB_PATH = "./mlflow/mlruns.sqlite" +$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 ` + --backend-store-uri="$BACKEND_URI" ` + --default-artifact-root="$DEFAULT_ARTIFACT_ROOT" ` + -p $MLFLOW_PORT diff --git a/run_mlflow_server.sh b/run_mlflow_server.sh index 39744b5..1d20505 100755 --- a/run_mlflow_server.sh +++ b/run_mlflow_server.sh @@ -1,8 +1,16 @@ -# Must be a relative path to be used in an option for mlflow server. -BACKEND_STORE_DB_PATH="./mlflow/mlruns.sqlite" -DEFAULT_ARTIFACTS_ROOT="./mlflow/" - -mlflow server \ - --backend-store-uri="sqlite:///$BACKEND_STORE_DB_PATH" \ - --default-artifacts-root="$DEFAULT_ARTIFACTS_ROOT" \ - -p 5000 +#!/bin/sh + +set -eu + +DEFAULT_ARTIFACTS_ROOT="./mlflow/mlartifacts/" + +: "${MLFLOW_PORT:=5000}" + +BACKEND_URI="sqlite:///$BACKEND_STORE_DB_PATH" + +mkdir -p "${DEFAULT_ARTIFACT_ROOT}" + +exec mlflow server \ + --backend-store-uri="$BACKEND_URI" \ + --default-artifact-root="$DEFAULT_ARTIFACT_ROOT" \ + -p "$MLFLOW_PORT"