@@ -174,7 +174,7 @@ def eager_legacy_trace_op(
174174
175175 if op_type == 'load_combine' :
176176 assert len (outputs .keys ()) == 1
177- key = list ( outputs .keys ())[ 0 ]
177+ key = next ( iter ( outputs .keys ()))
178178 for j in range (len (returns )):
179179 returns [j ]._share_underline_tensor_to (outputs [key ][j ])
180180 return
@@ -200,12 +200,12 @@ def eager_legacy_trace_op(
200200 )
201201 elif isinstance (returns , list ):
202202 assert len (outputs .keys ()) == 1
203- key = list ( outputs .keys ())[ 0 ]
203+ key = next ( iter ( outputs .keys ()))
204204 for j in range (len (returns )):
205205 outputs [key ][j ].reconstruct_from_ (returns [j ], False )
206206 else :
207207 assert len (outputs .keys ()) == 1
208- key = list ( outputs .keys ())[ 0 ]
208+ key = next ( iter ( outputs .keys ()))
209209 if isinstance (outputs [key ], list ):
210210 outputs [key ][0 ].reconstruct_from_ (returns , False )
211211 else :
@@ -285,12 +285,12 @@ def eager_trace_op(
285285 outputs [retname ][0 ].reconstruct_from_ (returns [i ], False )
286286 elif isinstance (returns , list ):
287287 assert len (outputs .keys ()) == 1
288- key = list ( outputs .keys ())[ 0 ]
288+ key = next ( iter ( outputs .keys ()))
289289 for j in range (len (returns )):
290290 outputs [key ][j ].reconstruct_from_ (returns [j ], False )
291291 else :
292292 assert len (outputs .keys ()) == 1
293- key = list ( outputs .keys ())[ 0 ]
293+ key = next ( iter ( outputs .keys ()))
294294 if isinstance (outputs [key ], list ):
295295 outputs [key ][0 ].reconstruct_from_ (returns , False )
296296 else :
0 commit comments