Staging commit for hyp2f1 gradient tests

Author: ColtAllen

pytensor/tensor/math.py

@@ -1389,6 +1389,11 @@ def hyp2f1(a, b, c, z):
     """Gaussian hypergeometric function."""
 
 
+@scalar_elemwise
+def hyp2f1_der(a, b, c, z):
+    """Derivatives for Gaussian hypergeometric function."""
+
+
 @scalar_elemwise
 def j0(x):
     """Bessel function of the first kind of order 0."""
@@ -3138,6 +3143,7 @@ def matmul(x1: "ArrayLike", x2: "ArrayLike", dtype: Optional["DTypeLike"] = None
     "logaddexp",
     "logsumexp",
     "hyp2f1",
+    "hyp2f1_der",
 ]
 
 DEPRECATED_NAMES = [

tests/tensor/test_math_scipy.py

@@ -763,12 +763,22 @@ def test_deprecated_module():
     ),
 )
 
+_good_broadcast_pentanary_hyp2f1_der = dict(
+    normal=(
+        random_ranged(0, 1000, (2, 3)),
+        random_ranged(0, 1000, (2, 3)),
+        random_ranged(0, 1000, (2, 3)),
+        random_ranged(0, 0.5, (2, 3)),
+        integers_ranged(-1, 3, (2, 3)),
+    ),
+)
+
 TestHyp2F1Broadcast = makeBroadcastTester(
     op=at.hyp2f1,
     expected=expected_hyp2f1,
     good=_good_broadcast_quaternary_hyp2f1,
+    grad=_good_broadcast_quaternary_hyp2f1,
     eps=2e-10,
-    mode=mode_no_scipy,
 )
 
 TestHyp2F1InplaceBroadcast = makeBroadcastTester(
@@ -802,10 +812,15 @@ def test_deprecated_module():
     expected=expected_hyp2f1,
     good=_good_broadcast_quaternary_hyp2f1,
     eps=2e-10,
-    mode=mode_no_scipy,
     inplace=True,
 )
 
+TestHyp2F1DerBroadcast = makeBroadcastTester(
+    op=at.hyp2f1_der,
+    expected=expected_hyp2f1,
+    good=_good_broadcast_pentanary_hyp2f1_der,
+    eps=2e-10,
+)
 
 class TestBetaIncGrad:
     def test_stan_grad_partial(self):