Fix unstable tests when run concurrently (#1724)

tracel-ai · May 5, 2024 · f8994e0 · f8994e0
1 parent 152509c
commit f8994e0
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/crates/burn-core/src/optim/adagrad.rs b/crates/burn-core/src/optim/adagrad.rs
@@ -172,7 +172,7 @@ mod tests {
         BinFileRecorder::<FullPrecisionSettings>::default()
             .record(
                 optimizer.to_record(),
-                std::env::temp_dir().as_path().join("test_optim"),
+                std::env::temp_dir().as_path().join("test_optim_adagrad"),
             )
             .unwrap();
 

diff --git a/crates/burn-core/src/optim/adam.rs b/crates/burn-core/src/optim/adam.rs
@@ -205,7 +205,7 @@ mod tests {
         BinFileRecorder::<FullPrecisionSettings>::default()
             .record(
                 optimizer.to_record(),
-                std::env::temp_dir().as_path().join("test_optim"),
+                std::env::temp_dir().as_path().join("test_optim_adam"),
             )
             .unwrap();
 

diff --git a/crates/burn-core/src/optim/adamw.rs b/crates/burn-core/src/optim/adamw.rs
@@ -217,7 +217,10 @@ mod tests {
         let _linear = optimizer.step(LEARNING_RATE, linear, grads);
         let temp_dir = TempDir::new().unwrap();
         BinFileRecorder::<FullPrecisionSettings>::default()
-            .record(optimizer.to_record(), temp_dir.path().join("test_optim"))
+            .record(
+                optimizer.to_record(),
+                temp_dir.path().join("test_optim_adamw"),
+            )
             .unwrap();
 
         let state_optim_before = optimizer.to_record();

diff --git a/crates/burn-core/src/optim/rmsprop.rs b/crates/burn-core/src/optim/rmsprop.rs
@@ -334,7 +334,10 @@ mod tests {
         let _linear = optimizer.step(LEARNING_RATE, linear, grads);
         let temp_dir = TempDir::new().unwrap();
         BinFileRecorder::<FullPrecisionSettings>::default()
-            .record(optimizer.to_record(), temp_dir.path().join("test_optim"))
+            .record(
+                optimizer.to_record(),
+                temp_dir.path().join("test_optim_rmsprop"),
+            )
             .unwrap();
 
         let state_optim_before = optimizer.to_record();