eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: reduce dev dependencies

Browse Source
This commit is contained in:
Carson M.
2025-03-17 01:00:33 -05:00
parent 909e41f26a
commit fccbf4daf7
8 changed files with 1 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Cargo.toml
View File

@@ -111,8 +111,5 @@ num-complex = { version = "0.4", default-features = false, optional = true }
anyhow = "1.0" anyhow = "1.0"
ureq = { version = "3", default-features = false, features = [ "native-tls" ] } ureq = { version = "3", default-features = false, features = [ "native-tls" ] }
image = "0.25" 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" ] } tracing-subscriber = { version = "0.3", default-features = false, features = [ "env-filter", "fmt" ] }
glassbench = "0.4"
tokio = { version = "1.36", features = [ "test-util" ] } tokio = { version = "1.36", features = [ "test-util" ] }
tokio-test = "0.4.3"

2
src/session/mod.rs
View File

@@ -390,7 +390,7 @@ impl Session {
/// ``` /// ```
/// # use std::sync::Arc; /// # use std::sync::Arc;
/// # use ort::{session::{Session, run_options::RunOptions}, value::{Value, ValueType, TensorRef}, tensor::TensorElementType}; /// # 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 mut session = Session::builder()?.with_intra_threads(2)?.commit_from_file("tests/data/upsample.onnx")?;
/// let input = ndarray::Array4::<f32>::zeros((1, 64, 64, 3)); /// let input = ndarray::Array4::<f32>::zeros((1, 64, 64, 3));
/// let options = RunOptions::new()?; /// let options = RunOptions::new()?;

1
src/tensor/ndarray.rs
View File

@@ -39,7 +39,6 @@ where
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use ndarray::{arr1, arr2, arr3}; use ndarray::{arr1, arr2, arr3};
use test_log::test;
use super::*; use super::*;

1
tests/mnist.rs
View File

@@ -7,7 +7,6 @@ use ort::{
tensor::ArrayExtensions, tensor::ArrayExtensions,
value::TensorRef value::TensorRef
}; };
use test_log::test;
#[test] #[test]
fn mnist_5() -> ort::Result<()> { fn mnist_5() -> ort::Result<()> {

1
tests/squeezenet.rs
View File

@@ -12,7 +12,6 @@ use ort::{
tensor::ArrayExtensions, tensor::ArrayExtensions,
value::TensorRef value::TensorRef
}; };
use test_log::test;
#[test] #[test]
fn squeezenet_mushroom() -> ort::Result<()> { fn squeezenet_mushroom() -> ort::Result<()> {

1
tests/thread_manager.rs
View File

@@ -7,7 +7,6 @@ use std::{
}; };
use ort::environment::{GlobalThreadPoolOptions, ThreadManager, ThreadWorker}; use ort::environment::{GlobalThreadPoolOptions, ThreadManager, ThreadWorker};
use test_log::test;
struct ThreadStats { struct ThreadStats {
active_threads: AtomicUsize active_threads: AtomicUsize

1
tests/upsample.rs
View File

@@ -7,7 +7,6 @@ use ort::{
session::{Session, builder::GraphOptimizationLevel}, session::{Session, builder::GraphOptimizationLevel},
value::TensorRef value::TensorRef
}; };
use test_log::test;
fn load_input_image<P: AsRef<Path>>(name: P) -> RgbImage { fn load_input_image<P: AsRef<Path>>(name: P) -> RgbImage {
// Load image, converting to RGB format // Load image, converting to RGB format

1
tests/vectorizer.rs
View File

@@ -8,7 +8,6 @@ use ort::{
session::{Session, builder::GraphOptimizationLevel}, session::{Session, builder::GraphOptimizationLevel},
value::Tensor value::Tensor
}; };
use test_log::test;
#[test] #[test]
fn vectorizer() -> ort::Result<()> { fn vectorizer() -> ort::Result<()> {