Skip to content
Snippets Groups Projects
Commit eff7dc2f authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Raise error if no input set instead of just a warning.

parent a5e8976d
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!163Export refactor
...@@ -187,7 +187,7 @@ class ExportNode(ABC): ...@@ -187,7 +187,7 @@ class ExportNode(ABC):
self.attributes["in_height"][idx] = get_height(tensor) self.attributes["in_height"][idx] = get_height(tensor)
self.attributes["in_width"][idx] = get_width(tensor) self.attributes["in_width"][idx] = get_width(tensor)
else: else:
print(f"No input for {self.node.name()}") raise RuntimeError(f"No input for {self.node.name()} at input {idx}")
for idx, list_child_node_in_id in enumerate(self.node.outputs()): for idx, list_child_node_in_id in enumerate(self.node.outputs()):
self.outputs += [node_in_id[0] self.outputs += [node_in_id[0]
for node_in_id in list_child_node_in_id] for node_in_id in list_child_node_in_id]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment