python311Packages.jax: fix build on Darwin (#376874)

This commit is contained in:
Sandro 2025-01-27 16:05:49 +01:00 committed by GitHub
commit e57ede447b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -95,6 +95,8 @@ buildPythonPackage rec {
] ]
++ lib.optionals stdenv.hostPlatform.isDarwin [ ++ lib.optionals stdenv.hostPlatform.isDarwin [
# SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated! # 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/shape_poly_test.py::ShapePolyTest"
"--deselect tests/tree_util_test.py::TreeTest" "--deselect tests/tree_util_test.py::TreeTest"
]; ];
@ -150,6 +152,7 @@ buildPythonPackage rec {
] ]
++ lib.optionals stdenv.hostPlatform.isDarwin [ ++ lib.optionals stdenv.hostPlatform.isDarwin [
# SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated! # SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated!
# reported at: https://github.com/jax-ml/jax/issues/26106
"testInAxesPyTreePrefixMismatchError" "testInAxesPyTreePrefixMismatchError"
"testInAxesPyTreePrefixMismatchErrorKwargs" "testInAxesPyTreePrefixMismatchErrorKwargs"
"testOutAxesPyTreePrefixMismatchError" "testOutAxesPyTreePrefixMismatchError"