python312Packages.pytensor: re-enable working tests

2025-03-04 10:02:23 +01:00
parent 78c6d58d0f
commit ec45368e95
1 changed files with 71 additions and 104 deletions
@@ -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 = [