blockifier: fix conversion from nested feltts counts to optional nested int list

Author: AvivYossef-starkware

crates/blockifier/src/execution/contract_class.rs

@@ -205,6 +205,13 @@ impl From<&NestedFeltCounts> for NestedIntList {
     }
 }
 
+impl From<&NestedFeltCounts> for Option<NestedIntList> {
+    fn from(counts: &NestedFeltCounts) -> Self {
+        // If it's a single leaf, it means the original bytecode_segment_lengths was None
+        if counts.is_leaf() { None } else { Some(counts.into()) }
+    }
+}
+
 /// The resource used to run a contract function.
 #[cfg_attr(feature = "transaction_serde", derive(serde::Deserialize))]
 #[derive(Clone, Copy, Default, Debug, Eq, PartialEq, Serialize)]

crates/starknet_os_runner/src/classes_provider.rs

@@ -39,7 +39,7 @@ fn compiled_class_v1_to_casm(class: &CompiledClassV1) -> CasmContractClass {
         prime,
         compiler_version: String::new(),
         bytecode,
-        bytecode_segment_lengths: Some(class.bytecode_segment_felt_sizes().into()),
+        bytecode_segment_lengths: class.bytecode_segment_felt_sizes().into(),
         hints: Vec::new(),
         pythonic_hints: None,
         entry_points_by_type: (&class.entry_points_by_type).into(),