Skip to content

Commit c80236b

Browse files
committed
Assert that a operator descriptor exists
1 parent a1dc797 commit c80236b

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

Deeploy/DeeployTypes.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2761,12 +2761,12 @@ def parse(self, default_channels_first: bool = True) -> bool:
27612761

27622762
self.layerBinding: 'OrderedDict[str, ONNXLayer]' = OrderedDict()
27632763
for node in flatSchedule:
2764-
# TODO: Change this into an assert once we've refactored all the parsers
2765-
if node.op in self.operatorDescriptors:
2766-
desc = self.operatorDescriptors[node.op]
2767-
desc.canonicalize(node, self.graph.opset)
2768-
assert desc.check(node), \
2769-
f"[ERROR OP {node.op}] Node {node.name} is not a valid instance of {node.op} operator"
2764+
assert node.op in self.operatorDescriptors, \
2765+
f"[ERROR] Error parsing node {node.name}. There is no descriptor for operator {node.op}."
2766+
desc = self.operatorDescriptors[node.op]
2767+
desc.canonicalize(node, self.graph.opset)
2768+
assert desc.check(node), \
2769+
f"[ERROR] Node {node.name} is not a valid instance of {node.op} operator"
27702770

27712771
layer = self._mapNode(node)
27722772
if isinstance(layer, ONNXLayer):

0 commit comments

Comments
 (0)