Continued downstream_1288 in new fork. Added Black formatting.

Author: ColtAllen

pytensor/scalar/math.py

@@ -1632,4 +1632,4 @@ def c_code(self, *args, **kwargs):
         raise NotImplementedError()
 
 
-hyp2f1_der = Hyp2F1Der(upgrade_to_float, name="hyp2f1_der")
+hyp2f1_der = Hyp2F1Der(upgrade_to_float, name="hyp2f1_der")

tests/tensor/test_math_scipy.py

@@ -822,6 +822,7 @@ def test_deprecated_module():
     eps=2e-10,
 )
 
+
 class TestBetaIncGrad:
     def test_stan_grad_partial(self):
         # This test combines the following STAN tests:

tests/tensor/test_special.py

@@ -8,7 +8,15 @@
 from pytensor.compile.function import function
 from pytensor.configdefaults import config
 from pytensor.tensor import scalar, scalars
-from pytensor.tensor.special import LogSoftmax, Softmax, SoftmaxGrad, log_softmax, softmax, poch, factorial
+from pytensor.tensor.special import (
+    LogSoftmax,
+    Softmax,
+    SoftmaxGrad,
+    log_softmax,
+    softmax,
+    poch,
+    factorial,
+)
 from pytensor.tensor.type import matrix, tensor3, tensor4, vector
 from tests.tensor.utils import random_ranged
 from tests import unittest_tools as utt
@@ -140,9 +148,7 @@ def test_valid_axis(self):
             SoftmaxGrad(-4)(*x)
 
 
-@pytest.mark.parametrize(
-    "z, m", [random_ranged(0, 5, (2,)), random_ranged(0, 5, (2,))]
-)
+@pytest.mark.parametrize("z, m", [random_ranged(0, 5, (2,)), random_ranged(0, 5, (2,))])
 def test_poch(z, m):
 
     _z, _m = scalars("z", "m")
@@ -162,8 +168,7 @@ def test_factorial(n):
 
     actual_fn = function([_n], factorial(_n))
     actual = actual_fn(n)
-    
+
     expected = scipy_factorial(n)
 
     assert np.allclose(actual, expected)
-