Remove PyKzgResources. Reveal the state diff size. (#1341)

crates/native_blockifier/src/lib.rs

@@ -21,8 +21,8 @@ pub mod transaction_executor;
 use errors::{add_py_exceptions, UndeclaredClassHashError};
 use py_block_executor::PyBlockExecutor;
 use py_transaction_execution_info::{
-    PyBouncerInfo, PyCallInfo, PyKzgResources, PyOrderedEvent, PyOrderedL2ToL1Message,
-    PyTransactionExecutionInfo, PyVmExecutionResources,
+    PyBouncerInfo, PyCallInfo, PyOrderedEvent, PyOrderedL2ToL1Message, PyTransactionExecutionInfo,
+    PyVmExecutionResources,
 };
 use py_validator::PyValidator;
 use pyo3::prelude::*;
@@ -47,7 +47,6 @@ fn native_blockifier(py: Python<'_>, py_module: &PyModule) -> PyResult<()> {
     py_module.add_class::<PyStateDiff>()?;
     py_module.add_class::<PyTransactionExecutionInfo>()?;
     py_module.add_class::<PyValidator>()?;
-    py_module.add_class::<PyKzgResources>()?;
     py_module.add_class::<PyVmExecutionResources>()?;
     py_module.add_class::<PyBouncerInfo>()?;
     py_module.add_class::<StorageConfig>()?;

crates/native_blockifier/src/py_transaction_execution_info.rs

@@ -179,22 +179,15 @@ impl From<VmExecutionResources> for PyVmExecutionResources {
     }
 }
 
-#[pyclass]
-#[derive(Clone, Default)]
-pub struct PyKzgResources {
-    #[pyo3(get)]
-    // The number of felts needed to store the state diff, used for KZG data availability.
-    pub state_diff_size: usize,
-}
-
 #[pyclass]
 #[derive(Clone, Default)]
 pub struct PyBouncerInfo {
     #[pyo3(get)]
     // The number of felts needed to store L1<>L2 messages.
     pub message_segment_length: usize,
     #[pyo3(get)]
-    pub kzg_resources: PyKzgResources,
+    // The number of felts needed to store the state diff.
+    pub state_diff_size: usize,
     #[pyo3(get)]
     pub additional_os_resources: PyVmExecutionResources,
 }

crates/native_blockifier/src/transaction_executor.rs

@@ -23,7 +23,7 @@ use crate::py_block_executor::{into_block_context, PyGeneralConfig};
 use crate::py_state_diff::{PyBlockInfo, PyStateDiff};
 use crate::py_transaction::py_tx;
 use crate::py_transaction_execution_info::{
-    PyBouncerInfo, PyKzgResources, PyTransactionExecutionInfo, PyVmExecutionResources,
+    PyBouncerInfo, PyTransactionExecutionInfo, PyVmExecutionResources,
 };
 use crate::py_utils::PyFelt;
 
@@ -102,7 +102,7 @@ impl<S: StateReader> TransactionExecutor<S> {
                 )?;
                 let py_bouncer_info = PyBouncerInfo {
                     message_segment_length: 0,
-                    kzg_resources: PyKzgResources { state_diff_size: 0 },
+                    state_diff_size: 0,
                     additional_os_resources: PyVmExecutionResources::from(additional_os_resources),
                 };