From b5e3be03a0fcba21623db9b641d980b48d67afd9 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 24 Jun 2025 15:49:56 +0200
Subject: [PATCH] python3Packages.pytensor: 2.31.3 -> 2.31.4

Diff: https://github.com/pymc-devs/pytensor/compare/refs/tags/rel-2.31.3...rel-2.31.4

Changelog: https://github.com/pymc-devs/pytensor/releases/tag/rel-2.31.4
---
 .../python-modules/pytensor/default.nix       | 145 ++++++++----------
 1 file changed, 68 insertions(+), 77 deletions(-)

diff --git a/pkgs/development/python-modules/pytensor/default.nix b/pkgs/development/python-modules/pytensor/default.nix
index 1f1c456b99ef..61e040f5294c 100644
--- a/pkgs/development/python-modules/pytensor/default.nix
+++ b/pkgs/development/python-modules/pytensor/default.nix
@@ -33,7 +33,7 @@
 
 buildPythonPackage rec {
   pname = "pytensor";
-  version = "2.31.3";
+  version = "2.31.4";
   pyproject = true;
 
   src = fetchFromGitHub {
@@ -43,7 +43,7 @@ buildPythonPackage rec {
     postFetch = ''
       sed -i 's/git_refnames = "[^"]*"/git_refnames = " (tag: ${src.tag})"/' $out/pytensor/_version.py
     '';
-    hash = "sha256-tvK8UzJZvX9X2NKgqkyhi0ZzAb38Lu0ULze4L1Z3YfU=";
+    hash = "sha256-wHkEZqgnau8DaoOaSFg0Ma6EtjGLmc+y4fskNEyk7yg=";
   };
 
   build-system = [
@@ -82,81 +82,72 @@ buildPythonPackage rec {
     rm -rf pytensor
   '';
 
-  disabledTests =
-    [
-      # ValueError: dtype attribute is not a valid dtype instance
-      "test_AddDS"
-      "test_AddSD"
-      "test_add_sd"
-      "test_grad"
-      "test_rop"
-    ]
-    ++ 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_size_implied_by_broadcasted_parameters"
-      "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"
-    ];
+  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_size_implied_by_broadcasted_parameters"
+    "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"
+  ];
 
   disabledTestPaths = [
     # Don't run the most compute-intense tests