Merge pull request #333635 from NixOS/blackjax

python312Packages.blackjax: 1.2.2 -> 1.2.3

pkgs/development/python-modules/blackjax/default.nix

@@ -1,23 +1,30 @@
 {
   lib,
-  stdenv,
   buildPythonPackage,
   pythonOlder,
   fetchFromGitHub,
-  pytest-xdist,
-  pytestCheckHook,
+
+  # build-system
   setuptools-scm,
+
+  # dependencies
   fastprogress,
   jax,
   jaxlib,
   jaxopt,
   optax,
   typing-extensions,
+
+  # checks
+  pytestCheckHook,
+  pytest-xdist,
+
+  stdenv,
 }:
 
 buildPythonPackage rec {
   pname = "blackjax";
-  version = "1.2.2";
+  version = "1.2.3";
   pyproject = true;
 
   disabled = pythonOlder "3.9";
@@ -26,7 +33,7 @@ buildPythonPackage rec {
     owner = "blackjax-devs";
     repo = "blackjax";
     rev = "refs/tags/${version}";
-    hash = "sha256-W89L/3bpvecZdCpLCQOppev+fPc+SXd/T+zZLBH2wJs=";
+    hash = "sha256-f1piE79TLVLtIe9/DaLhXss/ifhU719nEylyl70SVJc=";
   };
 
   build-system = [ setuptools-scm ];

pkgs/development/python-modules/jaxopt/default.nix

@@ -86,13 +86,21 @@ buildPythonPackage rec {
       "test_dtype_consistency"
       # AssertionError: Array(0.01411963, dtype=float32) not less than or equal to 0.01
       "test_multiclass_logreg6"
+    ]
+    ++ lib.optionals stdenv.isDarwin [
+      # Fatal Python error: Aborted
+      "test_dtype_consistency"
+
+      # AssertionError (flaky numerical tests)
+      "test_logreg_with_intercept_manual_loop3"
+      "test_binary_logit_log_likelihood"
     ];
 
-  meta = with lib; {
+  meta = {
     homepage = "https://jaxopt.github.io";
     description = "Hardware accelerated, batchable and differentiable optimizers in JAX";
     changelog = "https://github.com/google/jaxopt/releases/tag/jaxopt-v${version}";
-    license = licenses.asl20;
-    maintainers = with maintainers; [ bcdarwin ];
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [ bcdarwin ];
   };
 }