fix(dark_market): fix change cwd logic

zama-ai · May 24, 2023 · 4eac45f · 4eac45f
1 parent ddb3451
commit 4eac45f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tfhe/examples/dark_market.rs b/tfhe/examples/dark_market.rs
@@ -342,7 +342,9 @@ fn test_volume_match_fhe(
     fhe_function: fn(&mut [RadixCiphertextBig], &mut [RadixCiphertextBig], &ServerKey),
 ) {
     let working_dir = std::env::current_dir().unwrap();
-    std::env::set_current_dir(working_dir.join("tfhe")).unwrap();
+    if working_dir.file_name().unwrap() != std::path::Path::new("tfhe") {
+        std::env::set_current_dir(working_dir.join("tfhe")).unwrap();
+    }
 
     println!("Generating keys...");
     let time = Instant::now();