fix(provers): fix types on trait implementors

taikoxyz · Jul 17, 2024 · 866722d · 866722d
1 parent f1af18d
commit 866722d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/provers/risc0/driver/src/lib.rs b/provers/risc0/driver/src/lib.rs
@@ -54,7 +54,7 @@ impl Prover for Risc0Prover {
         input: GuestInput,
         output: &GuestOutput,
         config: &ProverConfig,
-        write: &mut dyn IdWrite,
+        write: Option<&mut dyn IdWrite>,
     ) -> ProverResult<Proof> {
         let config = Risc0Param::deserialize(config.get("risc0").unwrap()).unwrap();
 
@@ -95,7 +95,7 @@ impl Prover for Risc0Prover {
         Ok(Risc0Response { proof: journal }.into())
     }
 
-    async fn cancel(key: &str, store: &mut dyn IdStore) -> ProverResult<()> {
+    async fn cancel(key: &str, store: Box<&mut dyn IdStore>) -> ProverResult<()> {
         Ok(())
     }
 }

diff --git a/provers/sgx/prover/src/lib.rs b/provers/sgx/prover/src/lib.rs
@@ -71,7 +71,7 @@ impl Prover for SgxProver {
         input: GuestInput,
         _output: &GuestOutput,
         config: &ProverConfig,
-        _store: &mut dyn IdWrite,
+        _store: Option<&mut dyn IdWrite>,
     ) -> ProverResult<Proof> {
         let sgx_param = SgxParam::deserialize(config.get("sgx").unwrap()).unwrap();
 
@@ -148,7 +148,7 @@ impl Prover for SgxProver {
         sgx_proof.map(|r| r.into())
     }
 
-    async fn cancel(_proof_key: ProofKey, _read: &mut dyn IdStore) -> ProverResult<()> {
+    async fn cancel(_proof_key: ProofKey, _read: Box<&mut dyn IdStore>) -> ProverResult<()> {
         Ok(())
     }
 }

diff --git a/provers/sp1/driver/src/lib.rs b/provers/sp1/driver/src/lib.rs
@@ -32,7 +32,7 @@ impl Prover for Sp1Prover {
         input: GuestInput,
         _output: &GuestOutput,
         _config: &ProverConfig,
-        write: &mut dyn IdWrite,
+        write: Option<&mut dyn IdWrite>,
     ) -> ProverResult<Proof> {
         // Write the input.
         let mut stdin = SP1Stdin::new();
@@ -70,7 +70,7 @@ impl Prover for Sp1Prover {
         .into())
     }
 
-    async fn cancel(_key: &str, _store: &mut dyn IdStore) -> ProverResult<()> {
+    async fn cancel(_key: &str, _store: Box<&mut dyn IdStore>) -> ProverResult<()> {
         Ok(())
     }
 }