File tree 2 files changed +3
-3
lines changed 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -266,7 +266,7 @@ def clone_with_new_inputs(
266
266
267
267
assert isinstance (inputs , list | tuple )
268
268
remake_node = False
269
- new_inputs : list [" Variable" ] = list (inputs )
269
+ new_inputs : list [Variable ] = list (inputs )
270
270
271
271
# Some Ops like Alloc require the node to always be rebuilt in non-strict mode
272
272
# as the output type depends on the input values and not just their types
@@ -1272,7 +1272,7 @@ def clone_node_and_cache(
1272
1272
return None
1273
1273
1274
1274
# Use a cached `Op` clone when available
1275
- new_op : "Op" | None = cast (Optional ["Op" ], clone_d .get (node .op ))
1275
+ new_op : Op | None = cast (Optional ["Op" ], clone_d .get (node .op ))
1276
1276
1277
1277
cloned_inputs : list [Variable ] = [cast (Variable , clone_d [i ]) for i in node .inputs ]
1278
1278
Original file line number Diff line number Diff line change @@ -1577,7 +1577,7 @@ def __init__(
1577
1577
"""
1578
1578
from pytensor .graph .rewriting .unify import convert_strs_to_vars
1579
1579
1580
- var_map : dict [str , " Var" ] = {}
1580
+ var_map : dict [str , Var ] = {}
1581
1581
self .in_pattern = convert_strs_to_vars (in_pattern , var_map = var_map )
1582
1582
self .out_pattern = convert_strs_to_vars (out_pattern , var_map = var_map )
1583
1583
self .values_eq_approx = values_eq_approx
You can’t perform that action at this time.
0 commit comments