chore: only check for witness once conversion ended.

spencer-tb · Oct 25, 2024 · 2c931cd · 2c931cd
1 parent a796b49
commit 2c931cd
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/ethereum_test_specs/blockchain.py b/src/ethereum_test_specs/blockchain.py
@@ -580,11 +580,14 @@ def generate_block_data(
                 )
             )
             transition_tool_output.alloc = previous_alloc
-            # TODO: hack for now, replace with actual witness output once available from t8n
-            transition_tool_output.witness = Witness(
-                verkle_proof=transition_tool_output.result.verkle_proof,
-                state_diff=transition_tool_output.result.state_diff,
-            )
+            if transition_tool_output.result.verkle_conversion_ended:
+                # TODO: hack for now, replace with actual witness output once available from t8n
+                transition_tool_output.witness = Witness(
+                    verkle_proof=transition_tool_output.result.verkle_proof,
+                    state_diff=transition_tool_output.result.state_diff,
+                )
+            else:
+                transition_tool_output.witness = None
 
         return (
             env,