diff --git a/pytensor/graph/replace.py b/pytensor/graph/replace.py index 3c07e21232..09f5cc7a73 100644 --- a/pytensor/graph/replace.py +++ b/pytensor/graph/replace.py @@ -314,4 +314,4 @@ def transform(var: Variable) -> Variable: def vectorize(*args, **kwargs): warnings.warn("vectorize was renamed to vectorize_graph", UserWarning) - return vectorize_node(*args, **kwargs) + return vectorize_graph(*args, **kwargs)