Merge pull request #434 from hotg-ai/refactor/optional-tflite

Abstract out the different types of node in a zipped Rune
hotg-ai · Jun 7, 2022 · 41b7bd1 · 41b7bd1
2 parents 8d672c4 + 93d939f
commit 41b7bd1
Show file tree

Hide file tree

Showing 6 changed files with 1,397 additions and 1,283 deletions.
diff --git a/crates/runtime/src/lib.rs b/crates/runtime/src/lib.rs
@@ -32,6 +32,7 @@ mod tensor;
 pub mod builtins;
 mod outputs;
 
+#[cfg(feature = "zune")]
 pub mod zune;
 
 pub use crate::{

diff --git a/crates/runtime/src/models/mod.rs b/crates/runtime/src/models/mod.rs
@@ -15,7 +15,7 @@ use crate::callbacks::{Model, ModelMetadata};
 ///
 /// Supported formats are:
 /// - TensorFlow Lite
-#[cfg_attr(not(feature = "tflite"), doc("(not supported)"))]
+#[cfg_attr(not(feature = "tflite"), doc = "(not supported)")]
 pub fn default_model_handler(
     _id: u32,
     meta: &ModelMetadata<'_>,