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: cleanup

Browse Source
This commit is contained in:
Carson M.
2024-12-26 17:05:33 -06:00
parent 7f626ae66d
commit 794c04190e
2 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/error.rs
View File

@@ -148,7 +148,7 @@ pub(crate) fn assert_non_null_pointer<T>(ptr: *const T, name: &'static str) -> R
/// Converts an [`ort_sys::OrtStatus`] to a [`Result`].
///
/// Note that this frees `status`!
pub unsafe fn status_to_result(status: *mut ort_sys::OrtStatus) -> Result<(), Error> {
pub(crate) unsafe fn status_to_result(status: *mut ort_sys::OrtStatus) -> Result<(), Error> {
if status.is_null() {
Ok(())
} else {

8
tests/thread_manager.rs
View File

@@ -1,5 +1,4 @@
use std::{
path::Path,
sync::{
Arc,
atomic::{AtomicUsize, Ordering}
@@ -7,12 +6,7 @@ use std::{
thread::{self, JoinHandle}
};
use image::{ImageBuffer, Luma, Pixel, imageops::FilterType};
use ort::{
environment::{GlobalThreadPoolOptions, ThreadManager, ThreadWorker},
inputs,
session::{Session, builder::GraphOptimizationLevel}
};
use ort::environment::{GlobalThreadPoolOptions, ThreadManager, ThreadWorker};
use test_log::test;
struct ThreadStats {