From ef352dc520f266cd8b0a752fcc621f1834a3774d Mon Sep 17 00:00:00 2001 From: Patrick Robotham Date: Tue, 2 Dec 2025 23:00:24 +1100 Subject: [PATCH] Fix broken import of sp_sum sp_sum has been moved in pytensor from pytensor.sparse.basic to pytensor.sparse.math The higher level module pytensor.sparse exports this function. --- pymc/distributions/multivariate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pymc/distributions/multivariate.py b/pymc/distributions/multivariate.py index f76a98546e..4c43e6cc5b 100644 --- a/pymc/distributions/multivariate.py +++ b/pymc/distributions/multivariate.py @@ -25,7 +25,8 @@ from pytensor.graph.basic import Apply, Variable from pytensor.graph.op import Op from pytensor.raise_op import Assert -from pytensor.sparse.basic import DenseFromSparse, sp_sum +from pytensor.sparse import sp_sum +from pytensor.sparse.basic import DenseFromSparse from pytensor.tensor import ( TensorConstant, TensorVariable,