File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 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(
266266
267267 assert isinstance (inputs , list | tuple )
268268 remake_node = False
269- new_inputs : list [" Variable" ] = list (inputs )
269+ new_inputs : list [Variable ] = list (inputs )
270270
271271 # Some Ops like Alloc require the node to always be rebuilt in non-strict mode
272272 # as the output type depends on the input values and not just their types
@@ -1272,7 +1272,7 @@ def clone_node_and_cache(
12721272 return None
12731273
12741274 # 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 ))
12761276
12771277 cloned_inputs : list [Variable ] = [cast (Variable , clone_d [i ]) for i in node .inputs ]
12781278
Original file line number Diff line number Diff line change @@ -1577,7 +1577,7 @@ def __init__(
15771577 """
15781578 from pytensor .graph .rewriting .unify import convert_strs_to_vars
15791579
1580- var_map : dict [str , " Var" ] = {}
1580+ var_map : dict [str , Var ] = {}
15811581 self .in_pattern = convert_strs_to_vars (in_pattern , var_map = var_map )
15821582 self .out_pattern = convert_strs_to_vars (out_pattern , var_map = var_map )
15831583 self .values_eq_approx = values_eq_approx
You can’t perform that action at this time.
0 commit comments