diff --git a/pkgs/development/python-modules/pytensor/default.nix b/pkgs/development/python-modules/pytensor/default.nix index 2458524493e3..de6cddf923c6 100644 --- a/pkgs/development/python-modules/pytensor/default.nix +++ b/pkgs/development/python-modules/pytensor/default.nix @@ -79,111 +79,78 @@ buildPythonPackage rec { rm -rf pytensor ''; - disabledTests = - [ - # benchmarks (require pytest-benchmark): - "test_elemwise_speed" - "test_fused_elemwise_benchmark" - "test_logsumexp_benchmark" - "test_minimal_random_function_call_benchmark" - "test_scan_multiple_output" - "test_vector_taps_benchmark" + disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [ + # pytensor.link.c.exceptions.CompileError: Compilation failed (return status=1) + "OpFromGraph" + "add" + "cls_ofg1" + "direct" + "multiply" + "test_AddDS" + "test_AddSD" + "test_AddSS" + "test_MulDS" + "test_MulSD" + "test_MulSS" + "test_NoOutputFromInplace" + "test_OpFromGraph" + "test_adv_sub1_sparse_grad" + "test_alloc" + "test_binary" + "test_borrow_input" + "test_borrow_output" + "test_cache_race_condition" + "test_check_for_aliased_inputs" + "test_clinker_literal_cache" + "test_csm_grad" + "test_csm_unsorted" + "test_csr_dense_grad" + "test_debugprint" + "test_ellipsis_einsum" + "test_empty_elemwise" + "test_flatten" + "test_fprop" + "test_get_item_list_grad" + "test_grad" + "test_infer_shape" + "test_jax_pad" + "test_kron" + "test_masked_input" + "test_max" + "test_modes" + "test_mul_s_v_grad" + "test_multiple_outputs" + "test_not_inplace" + "test_numba_Cholesky_grad" + "test_numba_pad" + "test_optimizations_preserved" + "test_overided_function" + "test_potential_output_aliasing_induced_by_updates" + "test_profiling" + "test_rebuild_strict" + "test_runtime_broadcast_c" + "test_scan_err1" + "test_scan_err2" + "test_shared" + "test_solve_triangular_grad" + "test_structured_add_s_v_grad" + "test_structureddot_csc_grad" + "test_structureddot_csr_grad" + "test_sum" + "test_swap_SharedVariable_with_given" + "test_test_value_op" + "test_unary" + "test_unbroadcast" + "test_update_equiv" + "test_update_same" + ]; - # Failure reported upstream: https://github.com/pymc-devs/pytensor/issues/980 - "test_choose_signature" - - # AssertionError: Not equal to tolerance rtol=1e-06, atol=1e-06 - # Mismatched elements: 9 / 81 (11.1%) - "test_jax_pad" - ] - ++ lib.optionals stdenv.hostPlatform.isDarwin [ - # pytensor.link.c.exceptions.CompileError: Compilation failed (return status=1) - "OpFromGraph" - "add" - "cls_ofg1" - "direct" - "multiply" - "test_AddDS" - "test_AddSD" - "test_AddSS" - "test_MulDS" - "test_MulSD" - "test_MulSS" - "test_NoOutputFromInplace" - "test_OpFromGraph" - "test_adv_sub1_sparse_grad" - "test_alloc" - "test_binary" - "test_borrow_input" - "test_borrow_output" - "test_cache_race_condition" - "test_check_for_aliased_inputs" - "test_clinker_literal_cache" - "test_csm_grad" - "test_csm_unsorted" - "test_csr_dense_grad" - "test_debugprint" - "test_ellipsis_einsum" - "test_empty_elemwise" - "test_flatten" - "test_fprop" - "test_get_item_list_grad" - "test_grad" - "test_infer_shape" - "test_jax_pad" - "test_kron" - "test_masked_input" - "test_max" - "test_modes" - "test_mul_s_v_grad" - "test_multiple_outputs" - "test_not_inplace" - "test_numba_Cholesky_grad" - "test_numba_pad" - "test_optimizations_preserved" - "test_overided_function" - "test_potential_output_aliasing_induced_by_updates" - "test_profiling" - "test_rebuild_strict" - "test_runtime_broadcast_c" - "test_scan_err1" - "test_scan_err2" - "test_shared" - "test_solve_triangular_grad" - "test_structured_add_s_v_grad" - "test_structureddot_csc_grad" - "test_structureddot_csr_grad" - "test_sum" - "test_swap_SharedVariable_with_given" - "test_test_value_op" - "test_unary" - "test_unbroadcast" - "test_update_equiv" - "test_update_same" - ] - ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ - # Fatal Python error: Segmentation fault - # pytensor/link/basic.py", line 665 in thunk - "test_Unique" - "test_aligned_RandomVariable" - "test_blockwise" - "test_mvnormal_cov_decomposition_method" - "test_unnatural_batched_dims" - ]; - - disabledTestPaths = - [ - # Don't run the most compute-intense tests - "tests/scan/" - "tests/tensor/" - "tests/sparse/sandbox/" - ] - ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ - # Fatal Python error: Segmentation fault - # pytensor/link/basic.py", line 665 in thunk - "tests/link/numba/test_nlinalg.py" - "tests/link/numba/test_slinalg.py" - ]; + disabledTestPaths = [ + # Don't run the most compute-intense tests + "tests/scan/" + "tests/tensor/" + "tests/sparse/sandbox/" + ]; passthru.updateScript = nix-update-script { extraArgs = [