From 4665a66473018fe97ef41bb51d8e380356b52449 Mon Sep 17 00:00:00 2001
From: syropiatovvv <syrslava@yandex.ru>
Date: Sun, 2 Nov 2025 02:39:54 +0300
Subject: [PATCH] =?UTF-8?q?=D0=B4=D0=BE=D0=B1=D0=B0=D0=B2=D0=B8=D1=82?=
 =?UTF-8?q?=D1=8C=20=D1=81=D0=BA=D1=80=D0=B8=D0=BF=D1=82=D1=8B=20=D0=B4?=
 =?UTF-8?q?=D0=BB=D1=8F=20=D0=B2=D1=8B=D0=B7=D0=BE=D0=B2=20mlflow=20gc,=20?=
 =?UTF-8?q?=D0=B2=D1=8B=D0=BD=D0=B5=D1=81=D1=82=D0=B8=20=D0=BE=D0=B1=D1=89?=
 =?UTF-8?q?=D1=83=D1=8E=20=D1=87=D0=B0=D1=81=D1=82=D1=8C=20=D0=BA=D0=BE?=
 =?UTF-8?q?=D0=BD=D1=84=D0=B8=D0=B3=D1=83=D1=80=D0=B0=D1=86=D0=B8=D0=B8=20?=
 =?UTF-8?q?=D1=81=D0=BA=D1=80=D0=B8=D0=BF=D1=82=D0=BE=D0=B2=20=D0=B2=D1=8B?=
 =?UTF-8?q?=D0=B7=D0=BE=D0=B2=D0=B0=20mlflow=20=D0=B2=20=D0=BE=D1=82=D0=B4?=
 =?UTF-8?q?=D0=B5=D0=BB=D1=8C=D0=BD=D1=8B=D0=B9=20sourceable=20=D1=81?=
 =?UTF-8?q?=D0=BA=D1=80=D0=B8=D0=BF=D1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 _mlflow_config_common.ps1 | 2 ++
 _mlflow_config_common.sh  | 4 ++++
 gc_mlflow.ps1             | 5 +++++
 gc_mlflow.sh              | 7 +++++++
 run_mlflow_server.ps1     | 5 ++---
 run_mlflow_server.sh      | 6 +++---
 6 files changed, 23 insertions(+), 6 deletions(-)
 create mode 100644 _mlflow_config_common.ps1
 create mode 100644 _mlflow_config_common.sh
 create mode 100644 gc_mlflow.ps1
 create mode 100644 gc_mlflow.sh

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}"