From 87fa3c4e41ca74dc6e45b6b72124a711bc68c7d9 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 7 Oct 2024 17:10:18 +0200
Subject: [PATCH] mlflow-server: 2.14.3 -> 2.16.2

---
 pkgs/servers/mlflow-server/default.nix | 36 +++++++++++++-------------
 1 file changed, 18 insertions(+), 18 deletions(-)

diff --git a/pkgs/servers/mlflow-server/default.nix b/pkgs/servers/mlflow-server/default.nix
index 6916b3b65547..4e0d90e5dd6b 100644
--- a/pkgs/servers/mlflow-server/default.nix
+++ b/pkgs/servers/mlflow-server/default.nix
@@ -1,32 +1,32 @@
-{ python3, writeText}:
+{ python3Packages, writers}:
 
 let
-  py = python3.pkgs;
+  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: rec {
+  (py.mlflow.overridePythonAttrs(old: {
 
-    propagatedBuildInputs = old.propagatedBuildInputs ++ [
+    propagatedBuildInputs = old.dependencies ++ [
       py.boto3
       py.mysqlclient
     ];
 
     postPatch = (old.postPatch or "") + ''
-      substituteInPlace mlflow/utils/process.py --replace \
-        "child = subprocess.Popen(cmd, env=cmd_env, cwd=cwd, universal_newlines=True," \
-        "cmd[0]='$out/bin/gunicornMlflow'; child = subprocess.Popen(cmd, env=cmd_env, cwd=cwd, universal_newlines=True,"
-    '';
+      cat mlflow/utils/process.py
 
-    gunicornScript = writeText "gunicornMlflow"
-    ''
-        #!/usr/bin/env python
-        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())
-      '';
+      substituteInPlace mlflow/utils/process.py --replace-fail \
+        "process = subprocess.Popen(" \
+        "cmd[0]='${gunicornScript}'; process = subprocess.Popen("
+    '';
 
     postInstall = ''
       gpath=$out/bin/gunicornMlflow