From 8fe8da25dc6a5a61ca9796d904224b25876b27a1 Mon Sep 17 00:00:00 2001
From: SantiagoPittella <pittellasantiago@gmail.com>
Date: Mon, 24 Feb 2025 16:39:35 -0300
Subject: [PATCH] review: add instrumentation to batch prover

---
 crates/block-producer/src/batch_builder/mod.rs | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/crates/block-producer/src/batch_builder/mod.rs b/crates/block-producer/src/batch_builder/mod.rs
index 75e3c5bae..0e80a9b08 100644
--- a/crates/block-producer/src/batch_builder/mod.rs
+++ b/crates/block-producer/src/batch_builder/mod.rs
@@ -7,7 +7,7 @@ use miden_proving_service_client::proving_service::batch_prover::RemoteBatchProv
 use miden_tx_batch_prover::LocalBatchProver;
 use rand::Rng;
 use tokio::{task::JoinSet, time};
-use tracing::{debug, info, instrument, warn, Span};
+use tracing::{debug, info, instrument, Span};
 
 use crate::{
     domain::transaction::AuthenticatedTransaction, errors::BuildBatchError, mempool::SharedMempool,
@@ -314,17 +314,16 @@ impl BatchProver {
         Self::Remote(RemoteBatchProver::new(endpoint))
     }
 
+    #[instrument(target = COMPONENT, skip_all, err)]
     pub async fn prove(
         &self,
         proposed_batch: ProposedBatch,
     ) -> Result<ProvenBatch, BuildBatchError> {
         match self {
             Self::Local(prover) => {
-                info!(target: COMPONENT, "Proving batch locally");
                 prover.prove(proposed_batch).map_err(BuildBatchError::ProveBatchError)
             },
             Self::Remote(prover) => {
-                info!(target: COMPONENT, "Proving batch remotely");
                 prover.prove(proposed_batch).await.map_err(BuildBatchError::RemoteProverError)
             },
         }