diff --git a/pkgs/by-name/ml/mlflow-server/package.nix b/pkgs/by-name/ml/mlflow-server/package.nix index 8155f9b5c0c5..5f9a71fdf047 100644 --- a/pkgs/by-name/ml/mlflow-server/package.nix +++ b/pkgs/by-name/ml/mlflow-server/package.nix @@ -1,37 +1,10 @@ -{ python3Packages, writers }: +{ python3Packages }: -let - py = python3Packages; - - gunicornScript = writers.writePython3 "gunicornMlflow" { } '' - import re - import sys - from gunicorn.app.wsgiapp import run - if __name__ == '__main__': - sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', ''', sys.argv[0]) - sys.exit(run()) - ''; -in -py.toPythonApplication ( - py.mlflow.overridePythonAttrs (old: { - - propagatedBuildInputs = old.dependencies ++ [ - py.boto3 - py.mysqlclient +python3Packages.toPythonApplication ( + python3Packages.mlflow.overridePythonAttrs (old: { + propagatedBuildInputs = (old.propagatedBuildInputs or [ ]) ++ [ + python3Packages.boto3 + python3Packages.mysqlclient ]; - - 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(" - ''; - - postInstall = '' - gpath=$out/bin/gunicornMlflow - cp ${gunicornScript} $gpath - chmod 555 $gpath - ''; }) )