Raise warnings when test_val is accessed

Author: Dhruvanshu-Joshi

pytensor/graph/basic.py

@@ -479,7 +479,10 @@ def get_test_value(self):
         if not hasattr(self.tag, "test_value"):
             detailed_err_msg = get_variable_trace_string(self)
             raise TestValueError(f"{self} has no test value {detailed_err_msg}")
-
+        warnings.warn(
+            "test_value machinery is deprecated and will stop working in the future.",
+            FutureWarning,
+        )
         return self.tag.test_value
 
     def __str__(self):

pytensor/graph/fg.py

@@ -1,6 +1,7 @@
 """A container for specifying and manipulating a graph with distinct inputs and outputs."""
 
 import time
+import warnings
 from collections import OrderedDict
 from collections.abc import Iterable, Sequence
 from typing import TYPE_CHECKING, Any, Literal, Union, cast
@@ -493,6 +494,10 @@ def replace(
             return
 
         if config.compute_test_value != "off":
+            warnings.warn(
+                "test_value machinery is deprecated and will stop working in the future.",
+                FutureWarning,
+            )
             try:
                 tval = pytensor.graph.op.get_test_value(var)
                 new_tval = pytensor.graph.op.get_test_value(new_var)

pytensor/graph/op.py

@@ -69,6 +69,12 @@ def compute_test_value(node: Apply):
 
     """
     # Gather the test values for each input of the node
+
+    warnings.warn(
+        "compute_test_value is deprecated and will stop working in the future.",
+        FutureWarning,
+    )
+
     storage_map = {}
     compute_map = {}
     for i, ins in enumerate(node.inputs):
@@ -301,6 +307,10 @@ def __call__(
                     n.name = f"{name}_{i}"
 
         if config.compute_test_value != "off":
+            warnings.warn(
+                "test_value machinery is deprecated and will stop working in the future.",
+                FutureWarning,
+            )
             compute_test_value(node)
 
         if self.default_output is not None:
@@ -711,6 +721,11 @@ def get_test_values(*args: Variable) -> Any | list[Any]:
     if config.compute_test_value == "off":
         return []
 
+    warnings.warn(
+        "test_value machinery is deprecated and will stop working in the future.",
+        FutureWarning,
+    )
+
     rval = []
 
     for i, arg in enumerate(args):

pytensor/graph/rewriting/basic.py

@@ -983,6 +983,10 @@ def transform(self, fgraph, node, *args, **kwargs):
             if isinstance(input, pytensor.compile.SharedVariable):
                 pass
             elif hasattr(input.tag, "test_value"):
+                warnings.warn(
+                    "compute_test_value is deprecated and will stop working in the future.",
+                    FutureWarning,
+                )
                 givens[input] = pytensor.shared(
                     input.type.filter(input.tag.test_value),
                     input.name,

pytensor/misc/pkl_utils.py

@@ -9,6 +9,7 @@
 import pickle
 import sys
 import tempfile
+import warnings
 import zipfile
 from collections import defaultdict
 from contextlib import closing
@@ -61,6 +62,10 @@ class StripPickler(Pickler):
     def __init__(self, file, protocol=0, extra_tag_to_remove=None):
         # Can't use super as Pickler isn't a new style class
         super().__init__(file, protocol)
+        warnings.warn(
+            "compute_test_value is deprecated and will stop working in the future.",
+            FutureWarning,
+        )
         self.tag_to_remove = ["trace", "test_value"]
         if extra_tag_to_remove:
             self.tag_to_remove.extend(extra_tag_to_remove)

pytensor/scalar/basic.py

@@ -12,6 +12,7 @@
 
 import builtins
 import math
+import warnings
 from collections.abc import Callable, Mapping
 from copy import copy
 from itertools import chain
@@ -4414,6 +4415,10 @@ def apply(self, fgraph):
             if i.dtype == "float16":
                 mapping[i] = get_scalar_type("float32")()
                 if hasattr(i.tag, "test_value"):
+                    warnings.warn(
+                        "test_value machinery is deprecated and will stop working in the future.",
+                        FutureWarning,
+                    )
                     mapping[i].tag.test_value = i.tag.test_value
             else:
                 mapping[i] = i

pytensor/scan/basic.py

@@ -598,6 +598,10 @@ def wrap_into_list(x):
 
                 # Try to transfer test_value to the new variable
                 if config.compute_test_value != "off":
+                    warnings.warn(
+                        "test_value machinery is deprecated and will stop working in the future.",
+                        FutureWarning,
+                    )
                     try:
                         nw_slice.tag.test_value = get_test_value(_seq_val_slice)
                     except TestValueError:
@@ -725,6 +729,10 @@ def wrap_into_list(x):
 
             # Try to transfer test_value to the new variable
             if config.compute_test_value != "off":
+                warnings.warn(
+                    "test_value machinery is deprecated and will stop working in the future.",
+                    FutureWarning,
+                )
                 try:
                     arg.tag.test_value = get_test_value(actual_arg)
                 except TestValueError:
@@ -780,6 +788,10 @@ def wrap_into_list(x):
 
                 # Try to transfer test_value to the new variable
                 if config.compute_test_value != "off":
+                    warnings.warn(
+                        "test_value machinery is deprecated and will stop working in the future.",
+                        FutureWarning,
+                    )
                     try:
                         nw_slice.tag.test_value = get_test_value(_init_out_var_slice)
                     except TestValueError:

pytensor/scan/op.py

@@ -46,6 +46,7 @@
 import dataclasses
 import logging
 import time
+import warnings
 from collections import OrderedDict
 from collections.abc import Callable, Iterable
 from copy import copy
@@ -2650,6 +2651,10 @@ def compute_all_gradients(known_grads):
             # fct add and we want to keep it for all Scan op.  This is
             # used in T_Scan.test_grad_multiple_outs_taps to test
             # that.
+            warnings.warn(
+                "test_value machinery is deprecated and will stop working in the future.",
+                FutureWarning,
+            )
             if info.as_while:
                 n = n_steps.tag.test_value
             else:

pytensor/scan/rewriting.py

@@ -2,6 +2,7 @@
 
 import copy
 import dataclasses
+import warnings
 from itertools import chain
 from sys import maxsize
 from typing import cast
@@ -305,6 +306,10 @@ def add_to_replace(y):
             pushed_out_node = nd.op.make_node(*new_inputs)
 
             if config.compute_test_value != "off":
+                warnings.warn(
+                    "test_value machinery is deprecated and will stop working in the future.",
+                    FutureWarning,
+                )
                 compute_test_value(pushed_out_node)
 
             # Step 2. Create variables to replace the old outputs of the node
@@ -511,6 +516,10 @@ def add_to_replace(y):
             nw_outer_node = nd.op.make_node(*outside_ins)
 
             if config.compute_test_value != "off":
+                warnings.warn(
+                    "test_value machinery is deprecated and will stop working in the future.",
+                    FutureWarning,
+                )
                 compute_test_value(nw_outer_node)
 
             # Step 2. Create variables for replacements
@@ -545,6 +554,10 @@ def add_to_replace(y):
             replace_with_out.append(new_outer)
 
             if hasattr(new_outer.tag, "test_value"):
+                warnings.warn(
+                    "test_value machinery is deprecated and will stop working in the future.",
+                    FutureWarning,
+                )
                 new_sh = new_outer.tag.test_value.shape
                 ref_sh = (outside_ins.tag.test_value.shape[0],)
                 ref_sh += nd.outputs[0].tag.test_value.shape
@@ -982,6 +995,10 @@ def attempt_scan_inplace(
                 new_lsi = inp.owner.op.make_node(*inp.owner.inputs)
 
                 if config.compute_test_value != "off":
+                    warnings.warn(
+                        "test_value machinery is deprecated and will stop working in the future.",
+                        FutureWarning,
+                    )
                     compute_test_value(new_lsi)
 
                 new_lsi_out = new_lsi.outputs

pytensor/scan/utils.py

@@ -3,6 +3,7 @@
 import copy
 import dataclasses
 import logging
+import warnings
 from collections import OrderedDict, namedtuple
 from collections.abc import Callable, Sequence
 from itertools import chain
@@ -74,6 +75,10 @@ def safe_new(
         nw_x.name = nw_name
         if config.compute_test_value != "off":
             # Copy test value, cast it if necessary
+            warnings.warn(
+                "test_value machinery is deprecated and will stop working in the future.",
+                FutureWarning,
+            )
             try:
                 x_test_value = get_test_value(x)
             except TestValueError:
@@ -104,6 +109,10 @@ def safe_new(
     # between test values, due to inplace operations for instance. This may
     # not be the most efficient memory-wise, though.
     if config.compute_test_value != "off":
+        warnings.warn(
+            "test_value machinery is deprecated and will stop working in the future.",
+            FutureWarning,
+        )
         try:
             nw_x.tag.test_value = copy.deepcopy(get_test_value(x))
         except TestValueError:

pytensor/tensor/blas.py

@@ -78,6 +78,7 @@
 import logging
 import os
 import time
+import warnings
 
 import numpy as np
 
@@ -1967,6 +1968,10 @@ def R_op(self, inputs, eval_points):
         test_values_enabled = config.compute_test_value != "off"
 
         if test_values_enabled:
+            warnings.warn(
+                "test_value machinery is deprecated and will stop working in the future.",
+                FutureWarning,
+            )
             try:
                 iv0 = pytensor.graph.op.get_test_value(inputs[0])
             except TestValueError:

pytensor/tensor/random/rewriting/basic.py

@@ -1,3 +1,4 @@
+import warnings
 from itertools import chain
 
 from pytensor.compile import optdb
@@ -108,6 +109,10 @@ def local_rv_size_lift(fgraph, node):
     new_node = node.op.make_node(rng, None, *dist_params)
 
     if config.compute_test_value != "off":
+        warnings.warn(
+            "test_value machinery is deprecated and will stop working in the future.",
+            FutureWarning,
+        )
         compute_test_value(new_node)
 
     return new_node.outputs
@@ -187,6 +192,10 @@ def local_dimshuffle_rv_lift(fgraph, node):
     new_node = rv_op.make_node(rng, new_size, *new_dist_params)
 
     if config.compute_test_value != "off":
+        warnings.warn(
+            "test_value machinery is deprecated and will stop working in the future.",
+            FutureWarning,
+        )
         compute_test_value(new_node)
 
     out = new_node.outputs[1]