From 7a6f8438aaf750380a9fff799ca81ff5c7e2ae43 Mon Sep 17 00:00:00 2001 From: Michael McGeagh Date: Wed, 2 Dec 2020 15:29:22 +0000 Subject: MLBEDSW-2066 Improve Exception messages Minor refactoring to use fstrings. Improve Error classes to correctly inherit the base class. Use existing exception classes instead of plain exceptions where it makes sense. Signed-off-by: Michael McGeagh Change-Id: I0941c04e91010da1db77299517a8e2d896371e77 --- ethosu/vela/tflite_writer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'ethosu/vela/tflite_writer.py') diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index f747d471..06026ba5 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -21,6 +21,7 @@ import numpy as np from flatbuffers import encode from flatbuffers.builder import UOffsetTFlags +from .errors import VelaError from .nn_graph import PassPlacement from .operation import Op from .tensor import MemType @@ -44,7 +45,7 @@ tflite_file_identifier = "TFL" + str(tflite_version) def FinishWithFileIdentifier(self, rootTable, fid): if fid is None or len(fid) != 4: - raise Exception("fid must be 4 chars") + raise VelaError("FileIdentifier must be 4 chars") flags = N.Uint8Flags prepSize = 4 -- cgit v1.2.1