Skip to content

Commit da08d73

Browse files
committed
fix
1 parent d7c89f2 commit da08d73

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

onnx_array_api/translate_api/builder_emitter.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,10 +167,13 @@ def _emit_node(self, **kwargs: Dict[str, Any]) -> List[str]:
167167
raise NotImplementedError("Graph attribute not supported yet.")
168168
args.append(f"{k}={vatt}")
169169

170-
outs = ", ".join(map(self._clean_result_name, outputs))
170+
cleaned_outputs = list(map(self._clean_result_name, outputs))
171+
outs = ", ".join(cleaned_outputs)
171172
inps = ", ".join(map(self._clean_result_name, inputs))
172173
op_type = self._emit_node_type(op_type, domain)
173-
sdomain = "" if not domain else f", domain={domain!r}"
174+
# Let's add output names to make it easier to debug.
175+
soutputs = f", outputs={cleaned_outputs}"
176+
sdomain = soutputs if not domain else f", domain={domain!r}{soutputs}"
174177
if args:
175178
sargs = ", ".join(args)
176179
if inps:

0 commit comments

Comments
 (0)