@@ -107,8 +107,8 @@ def _fix_graph_names(self, graph_like: ir.Graph | ir.Function) -> bool:
107107 scoped_used_node_names : list [set [str ]] = [set ()]
108108
109109 # Counters for generating unique names (using list to pass by reference)
110- value_counter = collections .Counter ()
111- node_counter = collections .Counter ()
110+ value_counter : collections . Counter [ str ] = collections .Counter ()
111+ node_counter : collections . Counter [ str ] = collections .Counter ()
112112
113113 def enter_graph (graph_like ) -> None :
114114 """Callback for entering a subgraph."""
@@ -182,7 +182,7 @@ def _process_value(
182182 value : ir .Value ,
183183 used_value_names : set [str ],
184184 seen_values : set [ir .Value ],
185- value_counter : collections .Counter ,
185+ value_counter : collections .Counter [ str ] ,
186186 ) -> bool :
187187 """Process a value only if it hasn't been processed before."""
188188 if value in seen_values :
@@ -202,7 +202,7 @@ def _process_value(
202202 return modified
203203
204204 def _assign_value_name (
205- self , value : ir .Value , used_names : set [str ], counter : collections .Counter
205+ self , value : ir .Value , used_names : set [str ], counter : collections .Counter [ str ]
206206 ) -> bool :
207207 """Assign a name to an unnamed value. Returns True if modified."""
208208 assert not value .name , (
@@ -215,7 +215,7 @@ def _assign_value_name(
215215 return True
216216
217217 def _assign_node_name (
218- self , node : ir .Node , used_names : set [str ], counter : collections .Counter
218+ self , node : ir .Node , used_names : set [str ], counter : collections .Counter [ str ]
219219 ) -> bool :
220220 """Assign a name to an unnamed node. Returns True if modified."""
221221 assert not node .name , (
@@ -228,7 +228,7 @@ def _assign_node_name(
228228 return True
229229
230230 def _fix_duplicate_value_name (
231- self , value : ir .Value , used_names : set [str ], counter : collections .Counter
231+ self , value : ir .Value , used_names : set [str ], counter : collections .Counter [ str ]
232232 ) -> bool :
233233 """Fix a value's name if it conflicts with existing names. Returns True if modified."""
234234 original_name = value .name
@@ -249,7 +249,7 @@ def _fix_duplicate_value_name(
249249 return True
250250
251251 def _fix_duplicate_node_name (
252- self , node : ir .Node , used_names : set [str ], counter : collections .Counter
252+ self , node : ir .Node , used_names : set [str ], counter : collections .Counter [ str ]
253253 ) -> bool :
254254 """Fix a node's name if it conflicts with existing names. Returns True if modified."""
255255 original_name = node .name
@@ -269,7 +269,7 @@ def _fix_duplicate_node_name(
269269
270270
271271def _find_and_record_next_unique_name (
272- preferred_name : str , used_names : set [str ], counter : collections .Counter
272+ preferred_name : str , used_names : set [str ], counter : collections .Counter [ str ]
273273) -> str :
274274 """Generate a unique name based on the preferred name and current counter."""
275275 new_name = preferred_name
0 commit comments