Reformat with black

Author: shreyas3156

pytensor/tensor/basic.py

@@ -269,7 +269,9 @@ def get_scalar_constant(v, elemwise=True, only_process_constants=False, max_recu
         data = v.data
         if data.ndim != 0:
             raise NotScalarConstantError()
-    return get_underlying_scalar_constant(v, elemwise, only_process_constants, max_recur)
+    return get_underlying_scalar_constant(
+        v, elemwise, only_process_constants, max_recur
+    )
 
 
 def get_underlying_scalar_constant(

pytensor/tensor/rewriting/elemwise.py

@@ -22,7 +22,12 @@
 )
 from pytensor.graph.rewriting.db import SequenceDB
 from pytensor.graph.utils import InconsistencyError, MethodNotDefined
-from pytensor.tensor.basic import MakeVector, alloc, cast, get_underlying_scalar_constant
+from pytensor.tensor.basic import (
+    MakeVector,
+    alloc,
+    cast,
+    get_underlying_scalar_constant,
+)
 from pytensor.tensor.elemwise import CAReduce, DimShuffle, Elemwise
 from pytensor.tensor.exceptions import NotScalarConstantError
 from pytensor.tensor.rewriting.basic import register_canonicalize, register_specialize

pytensor/tensor/rewriting/math.py

@@ -1493,7 +1493,9 @@ def investigate(node):
         and investigate(node.inputs[0].owner)
     ):
         try:
-            cst = get_underlying_scalar_constant(node.inputs[1], only_process_constants=True)
+            cst = get_underlying_scalar_constant(
+                node.inputs[1], only_process_constants=True
+            )
 
             res = zeros_like(node.inputs[0], dtype=dtype, opt=True)
 
@@ -2329,7 +2331,9 @@ def local_abs_merge(fgraph, node):
                 inputs.append(i.owner.inputs[0])
             elif isinstance(i, Constant):
                 try:
-                    const = get_underlying_scalar_constant(i, only_process_constants=True)
+                    const = get_underlying_scalar_constant(
+                        i, only_process_constants=True
+                    )
                 except NotScalarConstantError:
                     return False
                 if not (const >= 0).all():

pytensor/tensor/rewriting/subtensor.py

@@ -1729,7 +1729,10 @@ def local_join_subtensors(fgraph, node):
             if step is None:
                 continue
             try:
-                if get_underlying_scalar_constant(step, only_process_constants=True) != 1:
+                if (
+                    get_underlying_scalar_constant(step, only_process_constants=True)
+                    != 1
+                ):
                     return None
             except NotScalarConstantError:
                 return None

tests/tensor/test_basic.py

@@ -3423,10 +3423,10 @@ def test_None_and_NoneConst(self, only_process_constants):
                 None, only_process_constants=only_process_constants
             )
         assert (
-                get_underlying_scalar_constant(
+            get_underlying_scalar_constant(
                 NoneConst, only_process_constants=only_process_constants
             )
-                is None
+            is None
         )