Update rust/tvm-sys/src/datatype.rs

Co-authored-by: Nick Hynes <nhynes@berkeley.edu>
apache · May 8, 2020 · 6a472e1 · 6a472e1
1 parent d8a1640
commit 6a472e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rust/tvm-sys/src/datatype.rs b/rust/tvm-sys/src/datatype.rs
@@ -110,7 +110,7 @@ impl FromStr for DataType {
         }
 
         let mut type_lanes = type_str.split('x');
-        let typ = type_lanes.next().expect("Missing dtype");
+        let typ = type_lanes.next().ok_or(ParseTvmError::MissingDataType)?;
         let lanes = type_lanes
             .next()
             .map(|l| <u16>::from_str_radix(l, 10))