python3Packages.jaxopt: fix build

2026-03-24 23:45:23 +00:00
parent 8abc1f0826
commit 87256fe0bf
1 changed files with 17 additions and 0 deletions
@@ -33,6 +33,23 @@ buildPythonPackage (finalAttrs: {
    hash = "sha256-vPXrs8J81O+27w9P/fEFr7w4xClKb8T0IASD+iNhztQ=";
  };

+  postPatch =
+    # TypeError: LogisticRegression.__init__() got an unexpected keyword argument 'multi_class'
+    ''
+      substituteInPlace jaxopt/_src/test_util.py \
+        --replace-fail "multi_class=multiclass," ""
+    ''
+    # AttributeError: jax.experimental.enable_x64 was removed in JAX v0.9.0; use jax.enable_x64(True) instead.
+    + ''
+      substituteInPlace \
+        tests/zoom_linesearch_test.py \
+        tests/lbfgsb_test.py \
+        tests/lbfgs_test.py \
+        --replace-fail \
+          "jax.experimental.enable_x64()" \
+          "jax.enable_x64(True)"
+    '';
+
  build-system = [ setuptools ];

  dependencies = [