MLBEDSW-2528: MLCE-219: Custom operator pass through
- Fixed custom operator pass through
- Added error printing functions for operators and tensor
- Minor cleanup of custom exception handling
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Idf295df1e4c544381dc480244d880c32fb285e38
diff --git a/ethosu/vela/test/test_model_reader.py b/ethosu/vela/test/test_model_reader.py
index ee9a51e..23e7e90 100644
--- a/ethosu/vela/test/test_model_reader.py
+++ b/ethosu/vela/test/test_model_reader.py
@@ -26,15 +26,7 @@
model_reader.read_model("no_tflite_file.txt", model_reader.ModelReaderOptions())
-def test_read_model_corrupt_contents(tmpdir):
- # Tests read_model with a corrupt .tflite file
- fname = tmpdir.join("corrupt.tflite")
- fname.write("abcde1234")
- with pytest.raises(InputFileError):
- model_reader.read_model(fname.strpath, model_reader.ModelReaderOptions())
-
-
def test_read_model_file_not_found(tmpdir):
# Tests read_model with a .tflite file that does not exist
- with pytest.raises(InputFileError):
+ with pytest.raises(FileNotFoundError):
model_reader.read_model("non_existing.tflite", model_reader.ModelReaderOptions())