Add dprint method to Nodes

Author: ricardoV94

pytensor/graph/basic.py

@@ -75,6 +75,18 @@ def get_parents(self):
         """
         raise NotImplementedError()
 
+    def dprint(self, **kwargs):
+        """Debug print itself
+
+        Parameters
+        ----------
+        kwargs:
+            Optional keyword arguments to pass to debugprint function.
+        """
+        from pytensor.printing import debugprint
+
+        return debugprint(self, **kwargs)
+
 
 class Apply(Node, Generic[OpType]):
     """A `Node` representing the application of an operation to inputs.

tests/graph/test_basic.py

@@ -31,6 +31,7 @@
 )
 from pytensor.graph.op import Op
 from pytensor.graph.type import Type
+from pytensor.printing import debugprint
 from pytensor.tensor import constant
 from pytensor.tensor.math import max_and_argmax
 from pytensor.tensor.type import TensorType, iscalars, matrix, scalars, vector
@@ -869,3 +870,10 @@ def test_single_pass_per_node(self, mocker):
         assert len(inspect.call_args_list) == len(
             {a for ((a, b), kw) in inspect.call_args_list}
         )
+
+
+def test_dprint():
+    r1, r2 = MyVariable(1), MyVariable(2)
+    o1 = MyOp(r1, r2)
+    assert o1.dprint(file="str") == debugprint(o1, file="str")
+    assert o1.owner.dprint(file="str") == debugprint(o1.owner, file="str")