1  use ort::session::builder::SessionBuilder;
 2  
 3  #[derive(Debug, thiserror::Error)]
 4  pub enum OnnxError {
 5      #[error("Failed to load model: {0}")]
 6      ModelLoad(#[from] ort::Error),
 7  
 8      #[error("Failed to configure session: {0}")]
 9      SessionConfig(#[from] ort::Error<SessionBuilder>),
10  
11      #[error("Shape Error: {0}")]
12      Shape(#[from] ndarray::ShapeError),
13  }