Skip to content

Commit

Permalink
mlflow-server: format
Browse files Browse the repository at this point in the history
  • Loading branch information
GaetanLepage committed Oct 8, 2024
1 parent 87fa3c4 commit 8e03a09
Showing 1 changed file with 14 additions and 11 deletions.
25 changes: 14 additions & 11 deletions pkgs/servers/mlflow-server/default.nix
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
{ python3Packages, writers}:
{ python3Packages, writers }:

let
py = python3Packages;

gunicornScript = writers.writePython3 "gunicornMlflow" {} ''
gunicornScript = writers.writePython3 "gunicornMlflow" { } ''
import re
import sys
from gunicorn.app.wsgiapp import run
Expand All @@ -12,25 +12,28 @@ let
sys.exit(run())
'';
in
py.toPythonApplication
(py.mlflow.overridePythonAttrs(old: {
py.toPythonApplication (
py.mlflow.overridePythonAttrs (old: {

propagatedBuildInputs = old.dependencies ++ [
py.boto3
py.mysqlclient
];

postPatch = (old.postPatch or "") + ''
cat mlflow/utils/process.py
postPatch =
(old.postPatch or "")
+ ''
cat mlflow/utils/process.py
substituteInPlace mlflow/utils/process.py --replace-fail \
"process = subprocess.Popen(" \
"cmd[0]='${gunicornScript}'; process = subprocess.Popen("
'';
substituteInPlace mlflow/utils/process.py --replace-fail \
"process = subprocess.Popen(" \
"cmd[0]='${gunicornScript}'; process = subprocess.Popen("
'';

postInstall = ''
gpath=$out/bin/gunicornMlflow
cp ${gunicornScript} $gpath
chmod 555 $gpath
'';
}))
})
)

0 comments on commit 8e03a09

Please sign in to comment.