generic collect_balances (anza-xyz#2985)

abklabs · Nov 27, 2024 · 2267b36 · 2267b36
1 parent f971558
commit 2267b36
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
@@ -3538,12 +3538,12 @@ impl Bank {
 
     pub fn collect_balances(
         &self,
-        batch: &TransactionBatch<SanitizedTransaction>,
+        batch: &TransactionBatch<impl SVMMessage>,
     ) -> TransactionBalances {
         let mut balances: TransactionBalances = vec![];
         for transaction in batch.sanitized_transactions() {
             let mut transaction_balances: Vec<u64> = vec![];
-            for account_key in transaction.message().account_keys().iter() {
+            for account_key in transaction.account_keys().iter() {
                 transaction_balances.push(self.get_balance(account_key));
             }
             balances.push(transaction_balances);