1
1
import copy
2
- import warnings
3
2
from typing import TYPE_CHECKING , Generator , Optional , Sequence , Union , cast
4
3
5
4
import pytensor
@@ -23,7 +22,6 @@ def rewrite_graph(
23
22
include : Sequence [str ] = ("canonicalize" ,),
24
23
custom_rewrite : Optional ["GraphRewriter" ] = None ,
25
24
clone : bool = False ,
26
- custom_opt : Optional ["GraphRewriter" ] = None ,
27
25
** kwargs ,
28
26
) -> Union [Variable , Sequence [Variable ], FunctionGraph ]:
29
27
"""Easily apply rewrites to a graph.
@@ -62,14 +60,6 @@ def rewrite_graph(
62
60
query_rewrites = optdb .query (RewriteDatabaseQuery (include = include , ** kwargs ))
63
61
_ = query_rewrites .rewrite (fgraph )
64
62
65
- if custom_opt is not None :
66
- warnings .warn (
67
- "`custom_opt` is deprecated; use `custom_rewrite` instead." ,
68
- DeprecationWarning ,
69
- stacklevel = 2 ,
70
- )
71
- custom_rewrite = custom_opt
72
-
73
63
if custom_rewrite :
74
64
custom_rewrite .rewrite (fgraph )
75
65
@@ -248,28 +238,3 @@ def get_clients_at_depth(
248
238
else :
249
239
assert var .owner is not None
250
240
yield var .owner
251
-
252
-
253
- DEPRECATED_NAMES = [
254
- (
255
- "optimize_graph" ,
256
- "`optimize_graph` is deprecated: use `rewrite_graph` instead." ,
257
- rewrite_graph ,
258
- ),
259
- ]
260
-
261
-
262
- def __getattr__ (name ):
263
- """Intercept module-level attribute access of deprecated symbols.
264
-
265
- Adapted from https://stackoverflow.com/a/55139609/3006474.
266
-
267
- """
268
- from warnings import warn
269
-
270
- for old_name , msg , old_object in DEPRECATED_NAMES :
271
- if name == old_name :
272
- warn (msg , DeprecationWarning , stacklevel = 2 )
273
- return old_object
274
-
275
- raise AttributeError (f"module { __name__ } has no attribute { name } " )
0 commit comments