Commit 8114a813 authored by Ihar Hrachyshka's avatar Ihar Hrachyshka
Browse files

python311Packages.jax: fix build on Darwin 

Another test case fails with the same generic error:

nanobind::detail::nb_func_error_except(): exception could not be
translated!
parent 81b9a5f9

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

+3 −0
Original line number Diff line number Diff line
@@ -95,6 +95,8 @@ buildPythonPackage rec { 
    ]

    ++ lib.optionals stdenv.hostPlatform.isDarwin [

      # SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated!

      # reported at: https://github.com/jax-ml/jax/issues/26106

      "--deselect tests/pjit_test.py::PJitErrorTest::testAxisResourcesMismatch"

      "--deselect tests/shape_poly_test.py::ShapePolyTest"

      "--deselect tests/tree_util_test.py::TreeTest"

    ];
@@ -150,6 +152,7 @@ buildPythonPackage rec { 
    ]

    ++ lib.optionals stdenv.hostPlatform.isDarwin [

      # SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated!

      # reported at: https://github.com/jax-ml/jax/issues/26106

      "testInAxesPyTreePrefixMismatchError"

      "testInAxesPyTreePrefixMismatchErrorKwargs"

      "testOutAxesPyTreePrefixMismatchError"