Shifting print_value to pytensorf.py

Author: itsdivya1309

pymc/model/core.py

@@ -40,7 +40,6 @@
 from pytensor.compile import DeepCopyOp, get_mode
 from pytensor.compile.sharedvalue import SharedVariable
 from pytensor.graph.basic import Constant, Variable, graph_inputs
-from pytensor.printing import Print
 from pytensor.scalar import Cast
 from pytensor.tensor.elemwise import Elemwise
 from pytensor.tensor.random.op import RandomVariable
@@ -2246,12 +2245,3 @@ def normal_logp(value, mu, sigma):
     )
 
     return var
-
-
-def print_value(var, name=None):
-    """Print value of variable when it is computed during sampling.
-    This is likely to affect sampling performance.
-    """
-    if name is None:
-        name = var.name
-    return Print(name)(var)

pymc/pytensorf.py

@@ -40,6 +40,7 @@
 )
 from pytensor.graph.fg import FunctionGraph
 from pytensor.graph.op import Op
+from pytensor.printing import Print
 from pytensor.scalar.basic import Cast
 from pytensor.scan.op import Scan
 from pytensor.tensor.basic import _as_tensor_variable
@@ -275,6 +276,15 @@ def floatX(X):
         return np.asarray(X, dtype=pytensor.config.floatX)
 
 
+def print_value(var, name=None):
+    """Print value of variable when it is computed during sampling.
+    This is likely to affect sampling performance.
+    """
+    if name is None:
+        name = var.name
+    return Print(name)(var)
+
+
 _conversion_map = {"float64": "int32", "float32": "int16", "float16": "int8", "float8": "int8"}