summaryrefslogtreecommitdiff
path: root/candle-nn/tests/optim.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-nn/tests/optim.rs')
-rw-r--r--candle-nn/tests/optim.rs3
1 files changed, 1 insertions, 2 deletions
diff --git a/candle-nn/tests/optim.rs b/candle-nn/tests/optim.rs
index f1d3b3f5..673d0455 100644
--- a/candle-nn/tests/optim.rs
+++ b/candle-nn/tests/optim.rs
@@ -4,8 +4,7 @@ extern crate intel_mkl_src;
#[cfg(feature = "accelerate")]
extern crate accelerate_src;
-mod test_utils;
-use test_utils::{to_vec0_round, to_vec2_round};
+use candle::test_utils::{to_vec0_round, to_vec2_round};
use anyhow::Result;
use candle::{Device, Tensor, Var};