File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed
Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments