diff --git a/Cargo.toml b/Cargo.toml index e2233d6..f756340 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -111,8 +111,5 @@ num-complex = { version = "0.4", default-features = false, optional = true } anyhow = "1.0" ureq = { version = "3", default-features = false, features = [ "native-tls" ] } image = "0.25" -test-log = { version = "0.2", default-features = false, features = [ "trace" ] } tracing-subscriber = { version = "0.3", default-features = false, features = [ "env-filter", "fmt" ] } -glassbench = "0.4" tokio = { version = "1.36", features = [ "test-util" ] } -tokio-test = "0.4.3" diff --git a/src/session/mod.rs b/src/session/mod.rs index 0fecb9b..9092e2c 100644 --- a/src/session/mod.rs +++ b/src/session/mod.rs @@ -390,7 +390,7 @@ impl Session { /// ``` /// # use std::sync::Arc; /// # use ort::{session::{Session, run_options::RunOptions}, value::{Value, ValueType, TensorRef}, tensor::TensorElementType}; - /// # fn main() -> ort::Result<()> { tokio_test::block_on(async { + /// # fn main() -> ort::Result<()> { tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap().block_on(async { /// let mut session = Session::builder()?.with_intra_threads(2)?.commit_from_file("tests/data/upsample.onnx")?; /// let input = ndarray::Array4::::zeros((1, 64, 64, 3)); /// let options = RunOptions::new()?; diff --git a/src/tensor/ndarray.rs b/src/tensor/ndarray.rs index 90af8a1..994ca82 100644 --- a/src/tensor/ndarray.rs +++ b/src/tensor/ndarray.rs @@ -39,7 +39,6 @@ where #[cfg(test)] mod tests { use ndarray::{arr1, arr2, arr3}; - use test_log::test; use super::*; diff --git a/tests/mnist.rs b/tests/mnist.rs index c97a266..3cde171 100644 --- a/tests/mnist.rs +++ b/tests/mnist.rs @@ -7,7 +7,6 @@ use ort::{ tensor::ArrayExtensions, value::TensorRef }; -use test_log::test; #[test] fn mnist_5() -> ort::Result<()> { diff --git a/tests/squeezenet.rs b/tests/squeezenet.rs index 7e90bc0..632ea0a 100644 --- a/tests/squeezenet.rs +++ b/tests/squeezenet.rs @@ -12,7 +12,6 @@ use ort::{ tensor::ArrayExtensions, value::TensorRef }; -use test_log::test; #[test] fn squeezenet_mushroom() -> ort::Result<()> { diff --git a/tests/thread_manager.rs b/tests/thread_manager.rs index aff06fb..6ae1c1b 100644 --- a/tests/thread_manager.rs +++ b/tests/thread_manager.rs @@ -7,7 +7,6 @@ use std::{ }; use ort::environment::{GlobalThreadPoolOptions, ThreadManager, ThreadWorker}; -use test_log::test; struct ThreadStats { active_threads: AtomicUsize diff --git a/tests/upsample.rs b/tests/upsample.rs index b656163..1a1a071 100644 --- a/tests/upsample.rs +++ b/tests/upsample.rs @@ -7,7 +7,6 @@ use ort::{ session::{Session, builder::GraphOptimizationLevel}, value::TensorRef }; -use test_log::test; fn load_input_image>(name: P) -> RgbImage { // Load image, converting to RGB format diff --git a/tests/vectorizer.rs b/tests/vectorizer.rs index aa01d08..0e9a045 100644 --- a/tests/vectorizer.rs +++ b/tests/vectorizer.rs @@ -8,7 +8,6 @@ use ort::{ session::{Session, builder::GraphOptimizationLevel}, value::Tensor }; -use test_log::test; #[test] fn vectorizer() -> ort::Result<()> {