Merge pull request #45 from nbigaouette/nbigaouette/issue44

Re-export ndarray
nbigaouette · Dec 26, 2020 · 694bb7c · 694bb7c
2 parents b24b4cd + 0cadee1
commit 694bb7c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/onnxruntime/examples/sample.rs b/onnxruntime/examples/sample.rs
@@ -1,9 +1,8 @@
 #![forbid(unsafe_code)]
 
-use ndarray::Array;
-
 use onnxruntime::{
-    environment::Environment, tensor::OrtOwnedTensor, GraphOptimizationLevel, LoggingLevel,
+    environment::Environment, ndarray::Array, tensor::OrtOwnedTensor, GraphOptimizationLevel,
+    LoggingLevel,
 };
 use tracing::Level;
 use tracing_subscriber::FmtSubscriber;

diff --git a/onnxruntime/src/lib.rs b/onnxruntime/src/lib.rs
@@ -132,6 +132,9 @@ pub mod tensor;
 pub use error::{OrtApiError, OrtError, Result};
 use sys::OnnxEnumInt;
 
+// Re-export ndarray as it's part of the public API anyway
+pub use ndarray;
+
 lazy_static! {
     // static ref G_ORT: Arc<Mutex<AtomicPtr<sys::OrtApi>>> =
     //     Arc::new(Mutex::new(AtomicPtr::new(unsafe {