aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite_reader.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/tflite_reader.py')
-rw-r--r--ethosu/vela/tflite_reader.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py
index 84c4c3c2..5667aff5 100644
--- a/ethosu/vela/tflite_reader.py
+++ b/ethosu/vela/tflite_reader.py
@@ -20,7 +20,6 @@ import os.path
import numpy as np
from .errors import InputFileError
-from .errors import UnsupportedFeatureError
from .nn_graph import Graph
from .nn_graph import Subgraph
from .operation import Operation
@@ -229,7 +228,7 @@ class TFLiteGraph:
def parse_operator_code(self, code):
c = code.BuiltinCode()
- if not c in builtin_operator_map:
+ if c not in builtin_operator_map:
msg = "The input file contains operator code {} which is currently not supported".format(c)
raise InputFileError(self.name, msg)
op_type, ser = builtin_operator_map[c]