summaryrefslogtreecommitdiff
path: root/candle-nn/tests/optim.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2023-08-27 09:42:22 +0100
committerGitHub <noreply@github.com>2023-08-27 09:42:22 +0100
commit5320aa6b7d339ff594d3886dd29634ea8cde6f17 (patch)
treeb8dc53eaea3966c288ac3c4a597b4d36c2deefa4 /candle-nn/tests/optim.rs
parenta8b39dd7b784b3c3cdd4d228813bd48b2d0d79bb (diff)
downloadcandle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.tar.gz
candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.tar.bz2
candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.zip
Move the test-utils bits to a shared place. (#619)
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};