152152 write_trace ,
153153)
154154
155- empty_scope : ScopeType = PdlDict ({"pdl_context" : DependentContext ([]), "__pdl_replay" : {}})
155+ empty_scope : ScopeType = PdlDict (
156+ {"pdl_context" : DependentContext ([]), "__pdl_replay" : {}}
157+ )
156158
157159
158160RefT = TypeVar ("RefT" )
@@ -632,8 +634,8 @@ def process_block_body_with_replay(
632634 if isinstance (block , LeafBlock ):
633635 block_id = block .pdl__id
634636 replay_scope = scope ["__pdl_replay" ]
635- assert ( isinstance (block_id , str ) )
636- assert ( isinstance (replay_scope , dict ) )
637+ assert isinstance (block_id , str )
638+ assert isinstance (replay_scope , dict )
637639 try :
638640 result = replay_scope [block_id ]
639641 background = SingletonContext (
@@ -645,12 +647,15 @@ def process_block_body_with_replay(
645647 yield_background (background )
646648 trace = block
647649 except KeyError :
648- result , background , scope , trace = process_block_body (state , scope , block , loc )
649- scope = scope | { "__pdl_replay" : (replay_scope | {block_id : result }) }
650+ result , background , scope , trace = process_block_body (
651+ state , scope , block , loc
652+ )
653+ scope = scope | {"__pdl_replay" : (replay_scope | {block_id : result })}
650654 else :
651655 result , background , scope , trace = process_block_body (state , scope , block , loc )
652656 return result , background , scope , trace
653657
658+
654659def process_block_body (
655660 state : InterpreterState ,
656661 scope : ScopeType ,
0 commit comments