Skip to content

Commit f8994e0

Browse files
authored
Fix unstable tests when run concurrently (#1724)
1 parent 152509c commit f8994e0

File tree

4 files changed

+10
-4
lines changed

4 files changed

+10
-4
lines changed

crates/burn-core/src/optim/adagrad.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ mod tests {
172172
BinFileRecorder::<FullPrecisionSettings>::default()
173173
.record(
174174
optimizer.to_record(),
175-
std::env::temp_dir().as_path().join("test_optim"),
175+
std::env::temp_dir().as_path().join("test_optim_adagrad"),
176176
)
177177
.unwrap();
178178

crates/burn-core/src/optim/adam.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ mod tests {
205205
BinFileRecorder::<FullPrecisionSettings>::default()
206206
.record(
207207
optimizer.to_record(),
208-
std::env::temp_dir().as_path().join("test_optim"),
208+
std::env::temp_dir().as_path().join("test_optim_adam"),
209209
)
210210
.unwrap();
211211

crates/burn-core/src/optim/adamw.rs

+4-1
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,10 @@ mod tests {
217217
let _linear = optimizer.step(LEARNING_RATE, linear, grads);
218218
let temp_dir = TempDir::new().unwrap();
219219
BinFileRecorder::<FullPrecisionSettings>::default()
220-
.record(optimizer.to_record(), temp_dir.path().join("test_optim"))
220+
.record(
221+
optimizer.to_record(),
222+
temp_dir.path().join("test_optim_adamw"),
223+
)
221224
.unwrap();
222225

223226
let state_optim_before = optimizer.to_record();

crates/burn-core/src/optim/rmsprop.rs

+4-1
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,10 @@ mod tests {
334334
let _linear = optimizer.step(LEARNING_RATE, linear, grads);
335335
let temp_dir = TempDir::new().unwrap();
336336
BinFileRecorder::<FullPrecisionSettings>::default()
337-
.record(optimizer.to_record(), temp_dir.path().join("test_optim"))
337+
.record(
338+
optimizer.to_record(),
339+
temp_dir.path().join("test_optim_rmsprop"),
340+
)
338341
.unwrap();
339342

340343
let state_optim_before = optimizer.to_record();

0 commit comments

Comments
 (0)