.wip

Author: ricardoV94

pytensor/scalar/basic.py

@@ -1302,19 +1302,7 @@ def __hash__(self):
     def __str__(self):
         if hasattr(self, "name") and self.name:
             return self.name
-        else:
-            param = [
-                (k, v)
-                for k, v in self.__dict__.items()
-                if k
-                not in ("name", "_op_use_c_code", "bool", "output_types_preference")
-            ]
-            if param:
-                classname = self.__class__.__name__
-                args = ", ".join(f"{k}={v}" for k, v in param)
-                return f"{classname}{{{args}}}"
-            else:
-                return self.__class__.__name__
+        return self.__class__.__name__
 
     def c_code_cache_version(self):
         return (4,)

tests/scalar/test_loop.py

@@ -285,6 +285,7 @@ def test_elemwise_inplace(mutate_arg_idx):
         [xv, yv],
         mode=get_default_mode().including("inplace"),
     )
+    fn.dprint()
     elem_op = fn.maker.fgraph.outputs[0].owner.op
     assert isinstance(elem_op, Elemwise) and isinstance(elem_op.scalar_op, ScalarLoop)
     destroy_map = elem_op.destroy_map

tests/tensor/test_math_scipy.py

@@ -994,6 +994,7 @@ def test_hyp2f1_grad_stan_cases(self):
         hyp2f1_out = pt.hyp2f1(a1, a2, b1, z)
         hyp2f1_grad = pt.grad(hyp2f1_out, [a1, a2, b1, z])
         f_grad = function([a1, a2, b1, z], hyp2f1_grad)
+        f_grad.dprint(print_memory_map=True)
 
         rtol = 1e-9 if config.floatX == "float64" else 2e-3
         for (
@@ -1128,6 +1129,7 @@ def test_benchmark(self, case, wrt, benchmark):
         hyp2f1_out = pt.hyp2f1(a1, a2, b1, z)
         hyp2f1_grad = pt.grad(hyp2f1_out, wrt=a1 if wrt == "a" else [a1, a2, b1, z])
         f_grad = function([a1, a2, b1, z], hyp2f1_grad)
+        f_grad.dprint(print_memory_map=True)
 
         (test_a1, test_a2, test_b1, test_z, *expected_dds) = case