fix: Ensure segment dir is in-scope during proving (#99)

risc0 · Mar 12, 2024 · d131901 · d131901
1 parent e7f6973
commit d131901
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/host/src/operations/mod.rs b/host/src/operations/mod.rs
@@ -311,6 +311,7 @@ pub fn prove_locally(
         encoded_input.len(),
         encoded_input.len() * 4 / 1_000_000
     );
+    let segment_dir = tempdir().unwrap();
 
     info!("Running the prover...");
     let session = {
@@ -332,8 +333,6 @@ pub fn prove_locally(
         let env = env_builder.build().unwrap();
         let mut exec = ExecutorImpl::from_elf(env, elf).unwrap();
 
-        let segment_dir = tempdir().unwrap();
-
         exec.run_with_callback(|segment| {
             Ok(Box::new(FileSegmentRef::new(&segment, segment_dir.path())?))
         })