diff options
author | 2018-02-05 15:14:20 -0800 | |
---|---|---|
committer | 2018-02-05 15:19:13 -0800 | |
commit | c8674c8a755525fb87073c0ac74dd21267a7f22f (patch) | |
tree | d3605e76f15b3c2c8eb6c008b07e608ced67715a /tensorflow/contrib/lite/java/src/test | |
parent | 1b19bc9edcd993e19e01cd935e9787e1672c4391 (diff) |
Verify tflite model in TFLite Java API
PiperOrigin-RevId: 184594561
Diffstat (limited to 'tensorflow/contrib/lite/java/src/test')
-rw-r--r-- | tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/NativeInterpreterWrapperTest.java | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/NativeInterpreterWrapperTest.java b/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/NativeInterpreterWrapperTest.java index 473f73816f..90323555d8 100644 --- a/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/NativeInterpreterWrapperTest.java +++ b/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/NativeInterpreterWrapperTest.java @@ -60,9 +60,7 @@ public final class NativeInterpreterWrapperTest { NativeInterpreterWrapper wrapper = new NativeInterpreterWrapper(INVALID_MODEL_PATH); fail(); } catch (IllegalArgumentException e) { - assertThat(e) - .hasMessageThat() - .contains("Model provided has model identifier ' is ', should be 'TFL3'"); + assertThat(e).hasMessageThat().contains("is not a valid flatbuffer model"); } } |